1

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

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

我想要的是：

创建一个新列表（列表或元组），其中所有可能的组合通过交叉匹配分组在一起（在列表或元组中），包括第五个（或第三个，如果它是元组+元组+布尔）元素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列表中来实现这一点。

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

4

1 回答 1

4

使用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}set 文字语法。

这会产生：

>>> 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)]
于 2012-10-06T19:34:53.383 回答