選擇滿足條件的元素

Question

我想從（很長）列表中選擇一個滿足條件的特定元素。 在我的特定情況下，列表中的每個項目都是一個類，而我想要的條件是例如item.id == 1。

我開始

[item for item in collection if item.id == 1]

但我不想遍歷整個列表，因為我敢肯定只有一個滿足條件的元素。

另一個選擇是

def check(collection):
    for item in collection:
        if item.id == 1:
            return item

有沒有更有效的方法？ Python 3是否為此提供了內置功能？ 另一個選擇可能是預處理列表以獲得字典，例如

{item.id : item for item in collection}

Answer 1

您可以使用生成器表達式和next()函數 ：

item = next((item for item in collection if item.id == 1), None)

這將找到集合中的第一個匹配元素，如果沒有匹配項，則返回None 。 它不會重復和評估任何超出找到一個元素的嚴格要求。

但是，如果您確實需要做很多事情，只需創建一次映射，然后查詢每次映射將更加有效：

mapping = {item.id: item for item in collection}
item = mapping.get(1)

因為這會將循環限制為僅一次，然后使用O（1）查找性能。 這是內存與速度之間的權衡。

Answer 2

要獲得滿足條件的第一個元素，請使用忍者的答案中概述的next() 。

為了獲得一個滿足條件的元素並檢查它是否唯一，我使用以下函數：

def single(seq):
    """Checks that the sequence has a single element and returns it."""
    """Analogous to .Net LINQ's Single()"""
    if hasattr(seq,"__len__"):
        assert len(seq)==1
        return seq[0]
    else:
        i=iter(seq)
        try: value=i.next()
        except StopIteration: raise AssertionError()
        try: i.next()
        except StopIteration: return value
        else: raise AssertionError()