如何根據條件壓縮列表元素？

Question

我有兩個表，說：

a = ['A', 'B', 'C'] 

和

b = ['B/X', 'Y/C', 'A/Z'] 

如何有效地壓縮名單合力得到：

[('A', 'A/Z'), ('B', 'B/X'), ('C', 'Y/C')] 

非常感謝您！

Answer 1

你可以簡單地先構造一個字典：

lookup = { ki : k for k in b for ki in k.split('/') } 

，然後使用列表理解：

[(k,lookup.get(k)) for k in a] 

這將在元組添加None作爲第二個元素，如果沒有的b元素有一個子元素匹配。它將生成：

>>> [(k,lookup.get(k)) for k in a] 
[('A', 'A/Z'), ('B', 'B/X'), ('C', 'Y/C')]