Come dividere un elenco ordinato in liste sub quando due differenza di valore vicini è più grande di una soglia

Question

ingresso:

• un sorted list, come questo: [1,2,3,8,10,15,16,17,18,22,23,27,30,31]
• una soglia, come questo: max_diff = 2

risultato atteso:

• una lista di liste sub; ciascuna lista sotto contiene i valori che la differenza vicini è più piccolo di max_diff, in questo modo: [[1, 2, 3], [8, 10], [15, 16, 17, 18], [22, 23], [27], [30, 31]]

Ecco come ho fatto questo, mi chiedo se c'è un modo migliore per farlo.

test_list = [1,2,3,8,10,15,16,17,18,22,23,27,30,31] 
max_diff = 2 

splited_list = [] 
temp_list = [test_list[0]] 
for i in xrange(1,len(test_list)): 
    if test_list[i] - temp_list[-1] > max_diff: 
     splited_list.append(temp_list) 
     temp_list = [test_list[i]] 
    else: 
     temp_list.append(test_list[i])   
    if i == len(test_list) -1: 
     splited_list.append(temp_list) 

print splited_list 

Answer 1

È possibile utilizzare enumerate e zip funzione all'interno di una lista di comprensione per trovare gli indici degli elementi che differenza valore è maggiore di 2, quindi dividere la lista in base a lista degli indici:

>>> li =[1, 2, 3, 8, 10, 15, 16, 17, 18, 22, 23, 27, 30, 31] 
>>> inds=[0]+[ind for ind,(i,j) in enumerate(zip(li,li[1:]),1) if j-i>2]+[len(li)+1] 
>>> [li[i:j] for i,j in zip(inds,inds[1:])] 
[[1, 2, 3], [8, 10], [15, 16, 17, 18], [22, 23], [27], [30, 31]] 

Answer 2

>>> a = [1,2,3,8,10,15,16,17,18,22,23,27,30,31] 
>>> b = a[1:] #offset by 1 position 
>>> b 
[2, 3, 8, 10, 15, 16, 17, 18, 22, 23, 27, 30, 31] 
>>> c = [(i[1] - i[0]) for i in zip(a[:-1], b)] 
>>> c #position diff 
[1, 1, 5, 2, 5, 1, 1, 1, 4, 1, 4, 3, 1] 
>>> d = [i[0] for i in enumerate(c) if i[1] > 2] 
>>> d #split position 
[2, 4, 8, 10, 11] 
>>> e = [-1]+d+[len(a)] 
>>> e #add start end to split position 
[-1, 2, 4, 8, 10, 11, 14] 
>>> [a[l[0]+1: l[1]+1] for l in zip(e, e[1:])] 
[[1, 2, 3], [8, 10], [15, 16, 17, 18], [22, 23], [27], [30, 31]] 
#split result 

Answer 3

Riorganizzare le linee conduce ad una forma più compatta:

test_list = [1,2,3,8,10,15,16,17,18,22,23,27,30,31] 
max_diff = 2 

splited_list = [] 
prev_element = float('-inf') 
for element in test_list: 
    if element - prev_element > max_diff: 
     splited_list.append([]) 
    splited_list[-1].append(element) 
    prev_element = element 
print splited_list 

Answer 4

Funziona su tutti i iterables

def split_by_threshold(seq, max_diff=2): 
    it = iter(seq) 
    last = next(it) 
    part = [last] 

    for curr in it: 
     if curr - last > max_diff: 
      yield part 
      part = [] 

     part.append(curr) 
     last = curr 

    yield part 


l = [1,2,3,8,10,15,16,17,18,22,23,27,30,31] 
print(list(split_by_threshold(l)))