Recherche d'une liste d'objets en Python

Question

Supposons que je crée une classe simple pour fonctionner de la même manière qu'une structure de type C, pour contenir uniquement des éléments de données. J'essaie de comprendre comment rechercher une liste d'objets pour des objets avec un attribut égal à une certaine valeur. Voici un exemple trivial pour illustrer ce que j'essaie de faire.

Par exemple:

class Data: 
    pass 

myList = [] 

for i in range(20): 
    data = Data() 
    data.n = i 
    data.n_squared = i * i 
    myList.append(data) 

Comment pourrais-je aller sur la recherche dans la liste myList pour déterminer si elle contient un élément avec n == 5?

J'ai été googling et recherche les docs Python, et je pense que je pourrais être en mesure de le faire avec une compréhension de la liste, mais je ne suis pas sûr. J'ajouterai que je dois utiliser Python 2.4.3 en passant, donc les nouvelles fonctionnalités de gee-whiz 2.6 ou 3.x ne sont pas disponibles pour moi.

Answer 1

Vous pouvez obtenir une liste de tous éléments correspondants avec une compréhension de la liste:

[x for x in myList if x.n == 30] # list of all elements with .n==30 

Si vous simplement vouloir déterminer si la liste contient un élément qui correspond et le faire (relativement) efficacement, vous pouvez le faire

def contains(list, filter): 
    for x in list: 
     if filter(x): 
      return True 
    return False 

if contains(myList, lambda x: x.n == 3) # True if any element has .n==3 
    # do stuff 

Answer 2

Vous devez ajouter un __eq__ et une méthode __hash__ à votre Data classe, il pourrait vérifier si les __dict__ attributs sont égaux (mêmes propriétés) et si leurs valeurs sont égales, aussi.

Si vous avez fait cela, vous pouvez utiliser

test = Data() 
test.n = 5 

found = test in myList 

Les chèques de mot-clé in si test est en myList.

Si vous voulez seulement aa propriété n à Data vous pouvez utiliser:

class Data(object): 
    __slots__ = ['n'] 
    def __init__(self, n): 
     self.n = n 
    def __eq__(self, other): 
     if not isinstance(other, Data): 
      return False 
     if self.n != other.n: 
      return False 
     return True 
    def __hash__(self): 
     return self.n 

    myList = [ Data(1), Data(2), Data(3) ] 
    Data(2) in myList #==> True 
    Data(5) in myList #==> False 

Answer 3

[x for x in myList if x.n == 30]    # list of all matches 
any(x.n == 30 for x in myList)     # if there is any matches 
[i for i,x in enumerate(myList) if x.n == 30] # indices of all matches 

def first(iterable, default=None): 
    for item in iterable: 
    return item 
    return default 

first(x for x in myList if x.n == 30)   # the first match, if any 

Answer 4

Juste pour être complet, il ne faut pas oublier la chose la plus simple qui pourrait travail:

for i in list: 
    if i.n == 5: 
    # do something with it 
    print "YAY! Found one!" 

Answer 5

filter(lambda x: x.n == 5, myList) 

Answer 6

Vous pouvez utiliser in pour rechercher un élément dans une collection, et liste de compréhension pour extraire le champ qui vous intéresse. Ceci (fonctionne pour les listes, les ensembles, les tuples, et tout ce qui définit __contains__ ou __getitem__).

if 5 in [data.n for data in myList]: 
    print "Found it" 

Voir aussi:

• Contains Method
• In operation

Answer 7

Simple, élégant et puissant:

Une expression du générateur en conjonction avec une commande interne ... (pyt L'honorable 2.5+)

any(x for x in mylist if x.n == 10) 

Utilise le Python any() builtin, qui est défini comme suit:

tout (itérables)-> Retour Vrai si un élément du itérables est vrai. Équivalent à:

def any(iterable): 
    for element in iterable: 
     if element: 
      return True 
    return False 

Answer 8

Tenir compte en utilisant un dictionnaire:

myDict = {} 

for i in range(20): 
    myDict[i] = i * i 

print(5 in myDict)