Python:如果包含特定属性值，则从命名元组列表中获取条目

Question

我有一个 list 的 namedtuples 如下

fruits = Fruits['type', 'color', 'weight', 'sweetness']
f1 = fruits('apple', 'red', 1.89, 5)
f1 = fruits('pear', 'green', 2.89, 7)
f1 = fruits('banana', 'yellow', 2.01, 10)

l = [f1, f2, f3]

现在，我想要一个函数，它从给定 type 的列表中返回一个特定的 namedtuple。我使用 for 循环编写了此函数，但是否可以做得更好(更快或不使用 for 循环)？

def take_fruit(type, all_fruits):
    for f in all_fruits:
        if f.type == type:
           return f
    return None

Answer 1

您可以使用 filter 或列表理解来使代码更短，但不一定更快:

def take_fruit_listcomp(type, all_fruits):
    try:
        return [f for f in all_fruits if f.type == type][0]
    except IndexError:
        return None

def take_fruit_filter(type, all_fruits):
    try:
        # no need for list(..) if you use Python 2
        return list(filter(lambda f: f.type == type, all_fruits))[0]
    except IndexError:
        return None