Cortar una lista en una lista de sub-listas

Question

¿Cuál es la forma más simple y razonablemente eficiente de dividir una lista en una lista de secciones de sub-listas segmentadas para sublistas de longitud arbitraria?

Por ejemplo, si nuestra lista de fuentes es:

input = [1, 2, 3, 4, 5, 6, 7, 8, 9, ... ] 

Y nuestro longitud de la lista de sub es 3, entonces buscamos:

output = [ [1, 2, 3], [4, 5, 6], [7, 8, 9], ... ] 

Del mismo modo, si nuestra longitud de la lista de sub es 4, entonces buscamos:

output = [ [1, 2, 3, 4], [5, 6, 7, 8], ... ] 

Answer 1

[input[i:i+n] for i in range(0, len(input), n)]  # use xrange in py2k 

donde n es la longitud de un fragmento.

Dado que no se define qué podría pasar con el elemento final de la nueva lista cuando el número de elementos en input no es divisible por n, asumí que no tiene importancia: con esto obtendrá el último elemento igual a 2 si n es igual a 7, por ejemplo.

Answer 2

La documentación del módulo itertools contiene la siguiente recipe:

import itertools 

def grouper(n, iterable, fillvalue=None): 
    "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx" 
    args = [iter(iterable)] * n 
    return itertools.izip_longest(fillvalue=fillvalue, *args) 

Esta función devuelve un iterador de tuplas de la longitud deseada:

>>> list(grouper(2, [1,2,3,4,5,6,7])) 
[(1, 2), (3, 4), (5, 6), (7, None)] 

Answer 3

me gusta la solución de SilentGhost.

Mi solución utiliza la programación funcional en Python:

group = lambda t, n: zip(*[t[i::n] for i in range(n)]) 
group([1, 2, 3, 4], 2) 

da:

[(1, 2), (3, 4)] 

Esto supone que el tamaño de la lista de entrada es divisible por el tamaño del grupo. Si no, los elementos no apareados no serán incluidos.

Answer 4

Una variante muy Pythonic (Python 3):

list(zip(*(iter([1,2,3,4,5,6,7,8,9]),)*3)) 

Se crea un iterador de lista y se convirtió en una tupla con 3x el mismo repetidor, a continuación, desempaquetado para comprimir y fundido a la lista de nuevo. Se extrae un valor de cada iterador por zip, pero como solo hay un único objeto iterador, el contador interno se incrementa globalmente para los tres.