Python交叉匹配

时间:2012-10-06 19:29:25

标签: python python-2.7

在执行某些算法之后,我最终得到一个类似下面的列表:

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 = {{ 1}})。

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

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]匹配一次,如果匹配元组是a或b是否为(l[1])或(l[0],l[1],..

现在我可以使用嵌套的l[1],l[0],..循环并弹出最后for e in l元素并执行e检查bool并创建新元组(或列表)并添加到{{ 1}}列表。

那我该怎么做呢?我该怎么办?

1 个答案:

答案 0 :(得分: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}集合文字语法。

这会产生:

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