Python交叉匹配

Question

在执行了一些算法之后，我最终得到了一个类似于下面的列表：

l = Set([(integer_4_digits, integer_n_digits], ..)
ex: l = Set([(1011, 123556), (1041, 424553), (1241, 464096), (1027, 589325), (1011, 432341), (1031, 423076)])
l = list(l) #all tuples must exist once. need to get items by index to match (perhaps?)

此列表中的每个元组的第一项是 4 位整数（正），第二项是另一个正整数。

我想要的是：

创建一个新列表（列表或元组），其中所有可能的组合通过交叉匹配组合在一起（在列表或元组中），包括第五个（或第三个，如果它是 tuple+tuple+bool）元素True 或 False 取决于两个新匹配的交叉元组的前 4 位整数是否相同（例如：1011 == 1011 所以 Bool=True）。

所以在这个过程之后，我想得到类似的东西：

new_list = [(l[0], l[1], False), (l[0], l[2], False), (l[0], l[3], False), (l[0], l[4], True) ..
            (l[1], l[2], False), (l[1], l[3], False), (l[1], l[4], False), ..
            (l[2], l[3], False), (l[2], l[4], False), (l[2], l[5], False), ..]

如您所见，new_list 不包含重复匹配项。 （l[0] 只匹配一次 l[1]，无论匹配元组是 a to be 还是 b to a (l[0],l[1],..) 或 (l[1],l[0],..)

现在我可以使用嵌套的for e in l 循环并弹出最后一个e 元素并执行e[0] 检查布尔并创建一个新元组（或列表）并添加到new_list 列表中来实现这一点。

那么我该怎么做呢？我该怎么做？

Answer 1

使用itertools.combinations();它会创建您正在寻找的组合。与列表推导一起，您应该设置：

from itertools import combinations

[(i, j, i[0] == j[0]) for i, j in combinations(l, 2)]

请注意，Set 类型是 python 中内置的原生类型；只需使用set（小写）作为本机版本。我们不需要把集合变成一个列表就可以了； itertools 不关心任何一种方式。您甚至可以使用新的 {elem, elem, elem} 设置文字语法。

这会产生：

>>> from itertools import combinations
>>> l = {(1011, 123556), (1041, 424553), (1241, 464096), (1027, 589325), (1011, 432341), (1031, 423076)}
>>> [(i, j, i[0] == j[0]) for i, j in combinations(l, 2)]
[((1041, 424553), (1027, 589325), False), ((1041, 424553), (1011, 123556), False), ((1041, 424553), (1031, 423076), False), ((1041, 424553), (1241, 464096), False), ((1041, 424553), (1011, 432341), False), ((1027, 589325), (1011, 123556), False), ((1027, 589325), (1031, 423076), False), ((1027, 589325), (1241, 464096), False), ((1027, 589325), (1011, 432341), False), ((1011, 123556), (1031, 423076), False), ((1011, 123556), (1241, 464096), False), ((1011, 123556), (1011, 432341), True), ((1031, 423076), (1241, 464096), False), ((1031, 423076), (1011, 432341), False), ((1241, 464096), (1011, 432341), False)]