在执行某些算法之后,我最终得到一个类似下面的列表:
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}}列表。
那我该怎么做呢?我该怎么办?
答案 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)]