在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)

使用Python any()内置函数，其定义如下：

any（iterable） -> 如果iterable中的任何元素为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"

也可以看看：

• 包含方法
• 运作中

您应该在类中添加__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)