Seleccione el valor de la lista de tuplas donde la condición

Question

Tengo una lista de tuplas. Cada tupla tiene 5 elementos (correspondientes a 5 columnas de base de datos) y me gustaría hacer una consulta

select attribute1 from mylist where attribute2 = something

por ejemplo,

personAge = select age from mylist where person_id = 10

¿Es posible consultar la lista de tuplas de alguna manera?

gracias

Answer 1

Si tiene named tuples se puede hacer esto:

results = [t.age for t in mylist if t.person_id == 10] 

utilizan de alguna manera índices:

results = [t[1] for t in mylist if t[0] == 10] 

O uso tupla desembalaje de acuerdo con la respuesta de Nate. Tenga en cuenta que no tiene que dar un nombre significativo a cada elemento que desempaquete. Puede hacer (person_id, age, _, _, _, _) para descomprimir una tupla de seis elementos.

Answer 2

Una solución a esto sería una lista por comprensión, la coincidencia de patrones dentro de la tupla:

>>> mylist = [(25,7),(26,9),(55,10)] 
>>> [age for (age,person_id) in mylist if person_id == 10] 
[55] 

Otra manera sería utilizar map y filter:

>>> map(lambda (age,_): age, filter(lambda (_,person_id): person_id == 10, mylist)) 
[55] 

Answer 3

Sí, puede utilizar el filtro si sabe en qué posición de la tupla reside la columna deseada. Si el caso es que el id es el primer elemento de la tupla a continuación, puede filtrar la lista de este modo:

filter(lambda t: t[0]==10, mylist) 

Esto devolverá la lista de tuplas correspondientes. Si quieres la edad, simplemente elige el elemento que quieras. En lugar de filtro, también puedes usar la comprensión de la lista y elegir el elemento en el primer intento. Incluso se puede desempaquetar inmediato (si sólo hay un resultado):

[age] = [t[1] for t in mylist if t[0]==10] 

pero recomiendo encarecidamente utilizar dictionaries o named tuples para este propósito.