Python - 匹配

Question

a = ('one', 'two')
b = ('ten', 'ten')

z = [('four', 'five', 'six'), ('one', 'two', 'twenty')]

我正在尝试 1) 查看我的元组中的前两个元素(例如 a 或 b)是否与我的元组列表 (z) 中的前两个元素匹配。 2)如果有匹配，我要返回元组的第三个元素

所以我想得到

myFunc(a,z) -> 'twenty'
myFunc(b,z) -> None

Answer 1

使用生成器表达式，和next :

>>> a = ('one', 'two')
>>> b = ('ten', 'ten')
>>> z = [('four', 'five', 'six'), ('one', 'two', 'twenty')]
>>> next((x[2] for x in z if x[:2] == a), None)
'twenty'
>>> next((x[2] for x in z if x[:2] == b), None)
>>>