comprensión de la lista de Python; ¿Comprimir una lista de listas?

Question

chicos. Intento encontrar la solución más elegante para un problema y me pregunto si Python tiene algo incorporado para lo que estoy tratando de hacer.

Lo que estoy haciendo es esto. Tengo una lista, A, y tengo una función f que toma un elemento y devuelve una lista. Puedo usar una lista de comprensión para convertir todo en A como tal;

[f(a) for a in A] 

Pero esto arroja una lista de listas;

[a1,a2,a3] => [[b11,b12],[b21,b22],[b31,b32]] 

Lo que realmente quiero es conseguir la lista aplanado;

[b11,b12,b21,b22,b31,b32] 

Ahora, otros idiomas tienen que; tradicionalmente se llama flatmap en lenguajes de programación funcionales, y .Net lo llama SelectMany. ¿Python tiene algo similar? ¿Hay una forma ordenada de asignar una función sobre una lista y aplanar el resultado?

El problema real que estoy tratando de resolver es esto; comenzando con una lista de directorios, encuentre todos los subdirectorios. asi que;

import os 
dirs = ["c:\\usr", "c:\\temp"] 
subs = [os.listdir(d) for d in dirs] 
print subs 

currentliy me da una lista de listas de tareas, pero realmente quiero una lista.

Answer 1

Puede haber anidado iteraciones en una lista única de comprensión:

[filename for path in dirs for filename in os.listdir(path)] 

Answer 2

Usted podría intentar itertools.chain(), así:

import itertools 
import os 
dirs = ["c:\\usr", "c:\\temp"] 
subs = list(itertools.chain(*[os.listdir(d) for d in dirs])) 
print subs 

itertools.chain() un iterador, por lo tanto, el paso a list().

Answer 3

subs = [] 
map(subs.extend, (os.listdir(d) for d in dirs)) 

(respuesta, pero de las hormigas es mejor; 1 para él)

Answer 4

Usted puede encontrar una buena respuesta en itertools' recipes:

def flatten(listOfLists): 
    return list(chain.from_iterable(listOfLists)) 

(Nota: requiere Python 2.6+)

Answer 5

Google me llevó siguiente solución:

def flatten(l): 
    if isinstance(l,list): 
     return sum(map(flatten,l)) 
    else: 
     return l 

Answer 6

Se podía hacer lo sencillo:

subs = [] 
for d in dirs: 
    subs.extend(os.listdir(d)) 

Answer 7

Puede concatenar listas utilizando el operador de suma normal:

>>> [1, 2] + [3, 4] 
[1, 2, 3, 4] 

La función integrada de sum agregará los números en una secuencia y puede comenzar opcionalmente de un valor específico:

>>> sum(xrange(10), 100) 
145 

Combine lo anterior para aplanar una lista de listas:

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

ahora puede definir su flatmap:

>>> def flatmap(f, seq): 
... return sum([f(s) for s in seq], []) 
... 
>>> flatmap(range, [1,2,3]) 
[0, 0, 1, 0, 1, 2] 

Edición: Acabo de ver la crítica en los comentarios de another answer y creo que es correcto que Python compilará innecesariamente y que la basura recopile muchas listas más pequeñas con esta solución. Así que lo mejor que se puede decir de él es que es muy simple y concisa, si estás acostumbrado a la programación funcional :-)

Answer 8

>>> listOfLists = [[1, 2],[3, 4, 5], [6]] 
>>> reduce(list.__add__, listOfLists) 
[1, 2, 3, 4, 5, 6] 

supongo itertools la solución es más eficiente que esto, pero esto se siente muy pitónico y evita tener que importar una biblioteca solo por el simple hecho de una operación de lista única.

Answer 9

import itertools 
x=[['b11','b12'],['b21','b22'],['b31']] 
y=list(itertools.chain(*x)) 
print y 

itertools trabajarán de python2.3 y una mayor

Answer 10

La pregunta propuesta flatmap. Se proponen algunas implementaciones pero pueden crear listas intermedias innecesarias. Aquí hay una implementación basada en iteradores.

def flatmap(func, *iterable): 
    return itertools.chain.from_iterable(map(func, *iterable)) 

In [148]: list(flatmap(os.listdir, ['c:/mfg','c:/Intel'])) 
Out[148]: ['SPEC.pdf', 'W7ADD64EN006.cdr', 'W7ADD64EN006.pdf', 'ExtremeGraphics', 'Logs'] 

En Python 2.x, utilice itertools.map en lugar de map.

Answer 11

If listA=[list1,list2,list3] 
flattened_list=reduce(lambda x,y:x+y,listA) 

Esto hará.

Answer 12

def flat_list(arr): 
    send_back = [] 
    for i in arr: 
     if type(i) == list: 
      send_back += flat_list(i) 
     else: 
      send_back.append(i) 
    return send_back 

Answer 13

Puede utilizar pyxtension:

from pyxtension.streams import stream 
stream([ [1,2,3], [4,5], [], [6] ]).flatMap() == range(7)