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'

我的问题是：有没有更好、更 Python 的方式来搜索这个列表？

Answer 1

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

1 如果我们想加快速度：

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

2 如果空间有限，请尝试其他答案的下一个或您的循环。

Answer 2

试试这个：

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

使用next() 的优点是我们只迭代查找所需内容所需的最少元素数量，所以不，它不等同于使用列表推导式创建一个全新的列表然后返回第一个元素。

Answer 3

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