Rechercher une liste d'objets en Python

Supposons que je crée une classe simple pour fonctionner de manière similaire à une structure de style C, pour contenir uniquement des éléments de données. J'essaie de comprendre comment rechercher une liste d'objets pour les 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 rechercher dans la liste myList pour déterminer si elle contient un élément avec n == 5?

J'ai cherché sur Google et recherché la documentation Python, et je pense que je pourrais peut-être le faire avec une compréhension de liste, mais je ne suis pas sûr. Je pourrais ajouter que je dois utiliser Python 2.4.3 en passant, donc toutes les nouvelles fonctionnalités de gee-whiz 2.6 ou 3.x ne sont pas disponibles pour moi.

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

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

Si vous voulez simplement déterminer si la liste contient un élément qui correspond et le faire (relativement) efficacement, vous pouvez 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

Simple, élégant et puissant:

Une expression de générateur en conjonction avec un builtin… (python 2.5+)

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

Utilise la fonction any()intégrée Python , qui est définie comme suit:

any (iterable) -> Renvoie True si un élément de l'itérable est vrai. Équivalent à:

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

Juste pour être complet, n'oublions pas la chose la plus simple qui pourrait éventuellement fonctionner:

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

[x for x in myList if x.n == 30]               # list of all matches
[x.n_squared for x in myList if x.n == 30]     # property of 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

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

Vous pouvez utiliser inpour rechercher un élément dans une collection et une compréhension de liste pour extraire le champ qui vous intéresse. Cela (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 également:

• Contient la méthode
• En opération

Vous devriez ajouter a __eq__et une __hash__méthode à votre Dataclasse, cela pourrait vérifier si les __dict__attributs sont égaux (mêmes propriétés) et ensuite si leurs valeurs sont égales aussi.

Si vous avez fait cela, vous pouvez utiliser

test = Data()
test.n = 5

found = test in myList

Le inmot-clé vérifie si testest dans myList.

Si vous ne voulez qu'une npropriété, Datavous 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

Pensez à utiliser un dictionnaire:

myDict = {}

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

print(5 in myDict)

Une autre façon de procéder consiste à utiliser la fonction next ().

matched_obj = next(x for x in list if x.n == 10)