Python - 从数组中选择一个具有某个属性的元素

Question

我已经遇到过这种情况几次了：我有一个数组，其中包含我想要的元素，并且我想在不知道索引的情况下选择该元素，而是知道一些期望的属性（也许这是一个字典列表，我想字典elem，使得elem['foo'] == 'bar'）。

我有一个解决方案是做一个过滤器（或更多的pythonically，列表理解），然后采取第一个元素（通常我知道被过滤的列表将是一个单身，所以采取第一个是唯一的） 。

Ex。给定一个列表x = [{'foo': 'bar_1'}, {'foo': 'bar_2'}]，我想要的元素的值为'foo'值为'bar2'。所以我这样做：

y = [elem for elem in x if elem['foo'] == 'bar_2'][0] 

有没有一个更标准的方式做到这一点？这似乎是一个非常简单和常见的用例。

Answer 1

如果您经常遇到此问题，请考虑使用不同的数据结构。想想一下，一张字典是否真的是你正在解决的问题的最佳数据结构。或者，如果是，请添加dict以跟踪您所需的信息。在这种情况下，为您打造x，还建立index：

index = {'bar2':x[1], 'bar1':x[0]} 

Dict lookups are O(1)，而任何基于列表的查找是O(n)。

Answer 2

可以使用generator呼吁它next拿到第一场比赛：

l = [{'foo': 'bar_1'}, {'foo': 'bar_2'}] 
print next(d for d in l if d["foo"] == "bar_2") 
{'foo': 'bar_2'} 

d = (d for d in l if d["foo"] == "bar_2") 
first = next(d) 

你也可以你itertools.dropwhile，如果你想前两个要素：

In [52]: from itertools import dropwhile,takewhile 
In [53]: l = [{'foo': 'bar_1'}, {'foo': 'bar_2'},{'foo': 'bar_2',"goo":"bar"}] 
In [54]: d = dropwhile(lambda x: x["foo"] != "bar_2",l) # drop elements whose where value of key "foo" is not equal to "bar_2" 
In [55]: first = next(d) 
In [56]: second = next(d)  
In [57]: first 
Out[57]: {'foo': 'bar_2'}  
In [58]: second 
Out[58]: {'foo': 'bar_2', 'goo': 'bar'}