python - 在 Python 中将两个列表拟合并对齐

Question

我正在尝试匹配由元组 (x,y,z) 组成的两个列表。我需要根据 z 值将它们组合在一起，并删除包含两个列表不常见的 z 值的任何元组。

输入:

a = [(0,0,0),(3,4,1),(5,3,2),(1,2,3)]
b = [(0,1,1),(2,3,2),(3,4,4)]

输出:

a1 = [(3,4,1),(5,3,2)]
b1 = [(0,1,1),(2,3,2)]

z 值将是整数，不重复。

Answer 1

common_z = set([tup[2] for tup in a]).intersection([tup[2] for tup in b])
a_ = [tup for tup in a if tup[2] in common_z]
b_ = [tup for tup in b if tup[2] in common_z]

>>> a_
[(3, 4, 1), (5, 3, 2)]

>>> b_
[(0, 1, 1), (2, 3, 2)]