6

次の問題があります。整数のリストを持っているので、元の入力リストの 2 つの要素の間のステップが 1 でないときはいつでも、それをリストのリストに分割したいと思います。例: input = [0, 1, 3, 5, 6, 7 ]、出力 = [[0、1]、[3]、[5、6、7]]

私は次の関数を書きましたが、それは地獄のように醜いです. itertools を使用しようとしましたが、解決できませんでした。

これが私の解決策です：

def _get_parts(list_of_indices):
    lv = list_of_indices
    tuples = zip(lv[:-1], lv[1:])
    split_values = []
    for i in tuples:
        if i[1] - i[0] != 1:
            split_values.append(i[1])
    string = '/'.join([str(i) for i in lv])
    substrings = []
    for i in split_values:
        part = string.split(str(i))
        substrings.append(part[0])
        string = string.lstrip(part[0])
    substrings.append(string)
    result = []
    for i in substrings:
        i = i.rstrip('/')
        result.append([int(n) for n in i.split('/')])
    return result

どうもありがとう！

4

5 に答える 5

8

これはあらゆるイテラブルで動作します

>>> from itertools import groupby, count
>>> inp = [0, 1, 3, 5, 6, 7]
>>> [list(g) for k, g in groupby(inp, key=lambda i,j=count(): i-next(j))]
[[0, 1], [3], [5, 6, 7]]
于 2013-02-22T08:40:35.000 に答える
2 
def _get_parts(i, step=1):
    o = []
    for x in i:
        if o and o[-1] and x - step == o[-1][-1]:
            o[-1].append(x)
        else:
            o.append([x])
    return o

_get_parts([0, 1, 3, 5, 6, 7], step=1)
# [[0, 1], [3], [5, 6, 7]])
于 2013-02-22T08:38:18.727 に答える
0

これは私がそれをする方法です：

inp = [0, 1, 3, 5, 6, 7]
base = []

for item in inp:
    if not base or item - base[-1][-1] != 1: # If base is empty (first item) or diff isn't 1
        base.append([item])                  # Append a new list containing just one item
    else:
        base[-1].append(item)                # Otherwise, add current item to the last stored list in base
print base                                   # => [[0, 1], [3], [5, 6, 7]]
于 2013-02-22T08:44:31.193 に答える
0

for ループを利用したソリューションを次に示します。

def splitbystep(alist):
  newlist = [[alist[0]]]
  for i in range(1,len(alist)):
    if alist[i] - alist[i-1] == 1:
      newlist[-1].append(alist[i])
    else:
      newlist.append([alist[i]])
  return newlist
于 2013-02-22T08:42:59.687 に答える