Erstellen Sie eine Liste von Tupeln mit benachbarten Zahlen?

Question

Ich habe eine Liste von Zahlen, zum Beispiel:

a = [1, 2, 3, 6, 9, 10] 

ich eine Funktion func schreiben möchten, die Liste und erzeugt Tupeln benachbarter Zahlen nimmt, so zum Beispiel die Liste wäre das Ergebnis

sein

func(a) == [(1,2,3), (6,), (9,10)] 

Was wäre ein effizienter Weg, dies zu tun?

Answer 1

Hier ist ein individueller Generator:

def yield_adjacent(lst): 
    it = iter(lst) 
    sub = [next(it)] 
    while True: 
     try: 
      n = next(it) 
     except StopIteration: 
      yield tuple(sub) 
      raise 
     if n - sub[-1] == 1: 
      sub.append(n) 
     else: 
      yield tuple(sub) 
      sub = [n] 

Demo:

>>> a = [1, 2, 3, 6, 9, 10] 
>>> list(yield_adjacent(a)) 
[(1, 2, 3), (6,), (9, 10)] 
>>> a = [] 
>>> list(yield_adjacent(a)) 
[] 
>>> a = [1, 3] 
>>> list(yield_adjacent(a)) 
[(1,), (3,)] 
>>> a = [1, 3, 4, 9] 
>>> list(yield_adjacent(a)) 
[(1,), (3, 4), (9,)] 

edit: for Schleife Version:

def yield_adjacent(lst): 
    it = iter(lst) 
    sub = [next(it)] 
    for n in it: 
     if n - sub[-1] == 1: 
      sub.append(n) 
     else: 
      yield tuple(sub) 
      sub = [n] 
    yield tuple(sub) 

Answer 2

Wenn Sie nach einem netten "funktionalen" Weg suchen, würde ich groupby in iwertools verwenden, wir müssen nur eine Funktion entwickeln, die einen Wert hat, wenn Elemente sequentiell sind und ein anderer, wenn sie nicht sind. Wir können nicht Zustand in der Funktion, die in groupby übergeben wird, so dass wir einen Verschluss verwenden:

from itertools import groupby 

def group_sequential(els): 
    # enumerate els to get index along with value 
    return [y for (x,y) in groupby(enumerate(els), lambda (idx,_): els[idx]-els[idx-1] if idx != 0 else 1)) 

bearbeiten: fixed Fehler in vorherigen Code.

Answer 3

Hier ist eine Funktion, die funktionieren soll:

def consec(l): 
    chunks = [i for i in range(1, len(l)) if l[i]-1 != l[i-1]]+[len(l)] 
    i = 0 
    for c in chunks: 
     yield(tuple(l[i:c])) 
     i = c