的Python：如果包含特定属性值

问题描述：

从一个名为元组的列表中的条目我有namedtuples一个list如下的Python：如果包含特定属性值

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]

现在，我想有从列表中返回一个特定namedtuple功能给出type。我使用for循环编写了这个函数，但它可以做得更好（更快还是没有for循环）？

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

'next（（f for all_fruits if for f.type == type），None）'但不知道它是否更快。 –

有几种方法可以做到这一点，但我认为没有比您的变体更快的方法。如果你想摆脱循环，你可以使用'过滤器'或列表理解。 – Zinki

答

您可以使用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

答

您可以使用此方法仅如果没有重复的类型。你可以为此使用字典。

d = { 
"apple" : fruits('apple', 'red', 1.89, 5), 
"pear" : fruits('pear', 'green', 2.89, 7), 
"banana" : fruits('banana', 'yellow', 2.01, 10) 
} 

def take_fruit(type) 
    return list(d[type])

这里，字典存储类型作为关键字。这种方式更快

如果有2个苹果怎么办？ – DeepSpace

在答案的开头添加了一个约束 – bigbounty

的Python：如果包含特定属性值

相关推荐