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 位整数(正)，第二项是另一个正整数。

我想要的是:

创建一个新列表(列表或元组)，其中所有可能的组合都通过交叉匹配组合在一起(在列表或元组中)，包括第五个(或第三个，如果它是元组+元组+ 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] 检查 bool 来实现这一点并创建一个新的元组(或列表)并添加到 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(小写)作为 native 版本。我们不需要将集合变成一个列表来工作； itertools 不关心任何一种方式。您甚至可以使用新的 {elem, elem, elem} set literal 语法。

这会产生:

>>> 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)]