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

試試這個：

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' 

注意，這會讓StopIteration異常，如果沒有找到，除非default提供：

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如果空間有限，嘗試其他答案的旁邊或你的循環。