Encuentra todas las subsecuencias consecutivas de longitud n en una secuencia

Quiero encontrar todas las subsecuencias consecutivas de longitud n en una secuencia.

Por ejemplo, digamos que n era 3 y la secuencia era:

[0,1,7,3,4,5,10] 

Quiero una función que produzca como salida:

 [[0,1,7],[1,7,3],[7,3,4],[3,4,5],[4,5,10]] 

¡Gracias por adelantado!

 >>> x = [0,1,7,3,4,5,10] >>> n = 3 >>> zip(*(x[i:] for i in range(n))) [(0, 1, 7), (1, 7, 3), (7, 3, 4), (3, 4, 5), (4, 5, 10)] 

Si desea que el resultado sea una lista de listas en lugar de una lista de tuplas, use map(list, zip(...)) .

 >>> x = [0,1,7,3,4,5,10] >>> [x[n:n+3] for n in range(len(x)-2)] [[0, 1, 7], [1, 7, 3], [7, 3, 4], [3, 4, 5], [4, 5, 10]] 
 def subseqs(seq, length): for i in xrange(len(seq) - length + 1): yield seq[i:i+length] 

Utilízalo como este:

 >>> for each in subseqs("hello", 3): ... print each ... hel ell llo 

Por supuesto funciona también con listas:

 >>> list(subseqs([1, 2, 3, 4, 5, 6, 7, 8], 3)) [[1, 2, 3], [2, 3, 4], [3, 4, 5], [4, 5, 6], [5, 6, 7], [6, 7, 8]] 

Lo siguiente probablemente sea adecuado para usted:

 def subseqs(xs, n): all_seqs = (xs[i:j+1] for i, _ in enumerate(xs) for j, _ in enumerate(xs)) return filter(lambda seq: len(seq) == n, all_seqs) >>> xs = [1, 2, 3, 4, 5, 6] # can be also range(1, 7) or list(range(1, 7)) >>> list(subseqs(xs, 3)) [[1, 2, 3], [2, 3, 4], [3, 4, 5], [4, 5, 6]] 

O simplemente, para obtener todas las secuencias de una lista llamada ‘xs’:

 [xs[i:j+1] for i, _ in enumerate(xs) for j, _ in enumerate(xs)] 

Para obtener las secuencias de una lista llamada ‘xs’ que son solo de longitud n:

 [xs[i:j+1] for i, _ in enumerate(xs) for j, _ in enumerate(xs) if len(xs[i:j+1]) == n]