Пошук у списку об’єктів у Python

Припустимо, я створюю простий клас для роботи, подібної до структури стилю C, щоб просто утримувати елементи даних. Я намагаюся зрозуміти, як шукати в списку об'єктів об'єкти з атрибутом, що дорівнює певному значенню. Нижче наведено тривіальний приклад для ілюстрації того, що я намагаюся зробити.

Наприклад:

class Data:
    pass

myList = []

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

Як би я розпочав пошук у списку myList, щоб визначити, чи містить він елемент з n == 5?

Я гуглював і шукав документи в Python, і думаю, що міг би це зробити із розумінням списку, але я не впевнений. Я можу додати, що мені до речі доводиться використовувати Python 2.4.3, тому будь-які нові функції gee-whiz 2.6 або 3.x мені недоступні.

Ви можете отримати список усіх відповідних елементів із розумінням списку:

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

Якщо ви просто хочете визначити, чи містить список будь-який елемент, який відповідає, і зробити це (відносно) ефективно, ви можете це зробити

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

Простий, елегантний та потужний:

Вираз генератора спільно з вбудованим… (python 2.5+)

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

Використовує any()вбудований Python , який визначається наступним чином:

any (iterable) -> Return True, якщо будь-який елемент iterable є true. Дорівнює:

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

Тільки для повноти, не забуваємо про найпростішу справу, яка могла б працювати:

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)

Ви можете використовувати inдля пошуку елемента в колекції та розуміння списку для вилучення поля, яке вас цікавить. Це (працює для списків, наборів, кортежів та всього, що визначає __contains__або __getitem__).

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

Дивіться також:

• Містить метод
• В експлуатації

Ви повинні додати a __eq__та __hash__метод у свій Dataклас, він може перевірити, чи є__dict__ атрибути (однакові властивості), а потім, чи рівні їх значення, теж.

Якщо ви зробили це, ви можете використовувати

test = Data()
test.n = 5

found = test in myList

У inключових словах перевіряє, testзнаходиться вmyList .

Якщо ви хочете лише мати nвластивість, Dataви можете використовувати:

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

Подумайте про використання словника:

myDict = {}

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

print(5 in myDict)

Інший спосіб зробити це - використовувати функцію next ().

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