python - 使用列表理解从元组中提取

Question

我有一个元组列表:

tuples = [(0,1), (2,0), (3,4), (1,2) etc. ]

我想创建另一个列表，其中包含所有已与 0 配对的数字。我尝试通过列表理解来执行此操作:

relations = [x[1] if x[0] == 0 else x[0] if x[1] == 0 for x in tuples]

但是，这会报错。 Python 似乎不喜欢将“x”作为元组。 “关系”可以用列表理解来定义吗？还是我需要写出更长的代码？

Answer 1

你还需要一个else;每个条件表达式都需要 if 和 else，您从第二个条件表达式中省略了 else。

但是您的问题可以更简单地表达为:

relations = [x[0] or x[1] for x in tuples if 0 in x]

这首先过滤其中包含零的任何元组，然后选择非 0 的值。

演示:

>>> tuples = [(0,1), (2,0), (3,4), (1,2)]
>>> [x[0] or x[1] for x in tuples if 0 in x]
[1, 2]