在Python中搜索对象列表

你可以得到一个列表 全部 将元素与列表理解进行匹配:

[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

简洁、优雅、强大:

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

使用Python any() 内置,定义如下:

-> 如果可迭代对象的任何元素为True,则返回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"

另请参见:

• Contains Method
• In operation

另一种方法是使用next()函数。

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

__eq__ 以及a __hash__ Data 类,它可以检查是否 __dict__ 属性是相等的(相同的属性),然后如果它们的值也是相等的。

如果你这样做了,你可以使用

test = Data()
test.n = 5

found = test in myList

这 in 关键字检查是否 test myList .

如果你只想a n 财产在 您可以使用:

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)

index 方法:

data_n = 30
j = [data.n for data in mylist].index(data_n)
print(mylist[j].data.n == data_n)