Python: chia danh sách các số nguyên dựa trên bước giữa chúng

Question

Tôi gặp sự cố sau. Có một danh sách các số nguyên, tôi muốn chia nó thành một danh sách các danh sách, bất cứ khi nào bước giữa hai phần tử của danh sách đầu vào gốc không phải là 1. Ví dụ: input = [0, 1, 3, 5, 6, 7], output = [[0, 1], [3], [5, 6, 7]]

Tôi đã viết hàm sau, nhưng nó là một địa ngục, và tôi tự hỏi liệu có ai trong số các bạn sẽ giúp tôi có được một giải pháp đẹp hơn. Tôi đã cố gắng sử dụng itertools, nhưng không thể giải quyết nó.

Đây là giải pháp của tôi:

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 

Cảm ơn rất nhiều!

Answer 1

này làm việc với bất kỳ iterable

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

Dưới đây là một giải pháp sử dụng một vòng lặp 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

Đây là cách tôi muốn làm điều đó:

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