Python: elenco diviso di interi in base al passaggio tra di loro

Question

Ho il seguente problema. Avendo una lista di interi, voglio dividerli, in una lista di liste, ogni volta che il passaggio tra due elementi della lista di input originale non è 1. Ad esempio: input = [0, 1, 3, 5, 6, 7], output = [[0, 1], [3], [5, 6, 7]]

Ho scritto la seguente funzione, ma è uggly come l'inferno, e mi chiedevo se qualcuno di voi ragazzi sarebbe aiutami a trovare una soluzione migliore. Ho provato ad usare itertools, ma non ho potuto risolverlo.

Ecco la mia soluzione:

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 

Grazie mille!

Answer 1

Questo funziona con qualsiasi iterabile

>>> 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]] 

Answer 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]]) 

Answer 3

Ecco una soluzione che utilizza un ciclo 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 

Answer 4

Ecco come lo farei:

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]]