Digamos que tengo una lista multidimensional l:conjunto de filtros para mostrar filas con un valor específico en una columna específica
l = [['a', 1],['b', 2],['c', 3],['a', 4]]
y quiero volver otra lista que consiste solamente en las filas que tiene 'a' en su primer elemento de lista:
m = [['a', 1],['a', 4]]
¿Cuál es una forma buena y eficiente de hacer esto?
Sin duda, un caso para una lista por comprensión:
m = [row for row in l if 'a' in row[0]]
Aquí estoy tomando el "tener 'a' en el primer elemento", literalmente, de ahí el uso del operador in. Si desea restringir esto a "tener 'a' como el primer elemento" (una cosa muy diferente de lo que realmente escribió -!), Entonces
m = [row for row in l if 'a' == row[0]]
es más como él ;-).
m = [i for i in l if i[0] == 'a']
Con la función filter:
m = filter(lambda x: x[0] == 'a', l)
o como una lista de comprensión:
m = [x for x in l where x[0] == 'a']
Qué le pasa a solo:
m = [i for i in l if i[0] == 'a']
O:
m = filter(lambda x: x[0] == 'a', l)
Dudo que la diferencia entre estos sea significativa en cuanto al rendimiento. Use el que sea más conveniente. No me gusta lambda s, pero el filter puede reemplazarse con itertools.ifilter para listas más grandes si eso es un problema, pero también puede cambiar la comprensión de la lista a un generador (cambie [] a ()) para lograr el mismo resultado general. Aparte de eso, probablemente sean idénticos.
[i for i in l if i[0]=='a']
por cierto, echar un vistazo a Pitón de list comprehension with conditions.