在Python中搜索对象列表

Question

假设我正在创建一个简单的类以类似于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功能。

Answer 1

您可以通过列表理解获得 所有 匹配元素的列表：

[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