trovare l'oggetto dal suo membro all'interno di una lista in Python

Question

lascia supporre la seguente semplice oggetto:

class Mock: 
    def __init__(self, name, age): 
     self.name = name 
     self.age = age 

poi ho una lista con alcuni oggetti di questo tipo:

myList = [Mock("Dan", 34), Mock("Jack", 30), Mock("Oli", 23)...] 

c'è qualche incorporato in funzione in cui posso ottenere tutti i mazzi con un'età di cioè 30? Certo che posso iterare me stesso su di loro e confrontare i loro età, ma qualcosa di simile

find(myList, age=30) 

sarebbe bello. C'è qualcosa del genere?

Answer 1

Si potrebbe desiderare di pre-index loro -

from collections import defaultdict 

class Mock(object): 
    age_index = defaultdict(list) 

    def __init__(self, name, age): 
     self.name = name 
     self.age = age 
     Mock.age_index[age].append(self) 

    @classmethod 
    def find_by_age(cls, age): 
     return Mock.age_index[age] 

Edit: una foto vale più di mille parole:

X axis è il numero di Mocks in myList, l'asse Y è runtime in secondi.

• puntini rossi sono @ filtro di dcrooney metodo()
• puntini blu sono @ di lista di marshall.ward
• punti verdi che si nascondono dietro l'asse X sono il mio indice ;-)

Answer 2

Si potrebbe provare un filter():

filter(lambda x: x.age == 30, myList) 

Questo sarebbe restituire una lista con solo gli oggetti che soddisfa l'espressione lambda.

Answer 3

di lista può prenderli:

new_list = [x for x in myList if x.age == 30] 

Answer 4

Lista le comprensioni sono quasi sempre il modo più veloce per fare queste cose (2x come veloce qui), anche se come accennato in precedenza l'indicizzazione è ancora più veloce se sei preoccupato per la velocità.

~$ python -mtimeit -s"from mock import myList" "filter(lambda x: x.age==21, myList)" 
1000000 loops, best of 3: 1.34 usec per loop 
~$ python -mtimeit -s"from mock import myList" "[x for x in myList if x.age==21]" 
1000000 loops, best of 3: 0.63 usec per loop 

Per file di mock.py nella directory corrente:

class Mock: 
    def __init__(self, name, age): 
     self.name = name 
     self.age = age 

myList = [Mock('Tom', 20), Mock('Dick', 21), Mock('Harry', 21), Mock('John', 22)]