Python:搜索元组列表

Question

我有一个元组列表，例如

>>> l = [ ("a",1), ("b",2), ("c",3) ]

我可以假设元素是独一无二的。现在我想得到元组的第一个元素，它的第二个元素是2（在本例中是'b'）。第一次尝试是：

>>> [ x for x, y in l if y == 2 ][0]
'b'

考虑到这会创建第二个列表来索引第0个元素，这似乎有些麻烦。另一种方法是反转给定列表l中的所有元组，并构建一个字典，然后为该字典编制索引：

>>> dict([ (y, x) for x, y in l ])[2]
'b'

考虑到反转列表和创建字典所涉及的数据混乱量，这似乎更为尴尬。最后，最详细但可能最快的方法是简单地迭代列表：

>>> def get(l) :
...     for x, y in l :
...         if y == 2 : 
...             return x
...     assert not "Should not happen."
... 
>>> get(l)
'b'

我的问题是：有没有更好、更像蟒蛇的方法来搜索这个列表？

Answer 1

试试这个：

next(x for x in l if x[1] == 2)[0]

使用next()的优点是，我们只对查找所需的最小元素量进行迭代，所以不，这不等同于使用列表理解创建一个全新的列表，然后返回第一个元素。

Answer 2

您也可以使用next():

In [1]: l = [("a", 1), ("b", 2), ("c", 3)]
In [2]: next(a for a, b in l if b == 2)
Out[2]: 'b'

请注意，如果未提供default，则会引发StopIteration异常：

In [3]: next(a for a, b in l if b == 100)
---------------------------------------------------------------------------
StopIteration                             
Traceback (most recent call last)
<ipython-input-38-14fe91d87aab> in <module>()
----> 1 next(a for a, b in l if b == 100)
StopIteration: 
In [4]: next((a for a, b in l if b == 100), 0)
Out[4]: 0

Answer 3

这取决于付款方式、空间或时间。你不可能两者兼得。

1如果我们想加快速度：

l = [ ("a",1), ("b",2), ("c",3) ]
_dict = {k:v for v,k in l}
print(_dict.get(2, None))

2如果空间有限，试试另一个答案的下一个或你的循环。