python - 如何根据原始元组列表中的值创建新的元组列表？

Question

我的函数当前正在返回:

[('a', 'b', 'c'), ('d', 'e', 'f'), ('g', 'e', 'f'), ('h', 'b', 'c')]

但是，我需要的最终输出是:

[('a', 'h'), ('d', 'g')]

如您所见，如果 i[1] 和 i[2] 匹配，我需要将 i[0] 配对在一起。

我正在尝试使用 for 循环，但此刻我想不出如何编写它。

Answer 1

这似乎可行:

from itertools import combinations

l = [('a', 'b', 'c'), ('d', 'e', 'f'), ('g', 'e', 'f'), ('h', 'b', 'c')]
print([(a[0], b[0]) for a, b in combinations(l, 2) if a[1:] == b[1:]])