Python生成带有条件的元组组合

时间:2018-08-17 20:11:20

标签: python-3.x tuples combinations

尝试生成一个组合,其中不包括c中提到的组合,并且不确定如何进行。

a = [1.01, 5.84, 13.86]

b = [6.42,0,18.24]

c = [(1.01,18.24), (13.86,0), (5.84,0)]

combination = [(x,y) for x in a 

                     for y in b]

预期结果: (1.01,6.42),(1.01,0),(5.84,6.42),(5.84,18.24),(13.86,6.42),(13.86,18.24)

1 个答案:

答案 0 :(得分:1)

您可以使用itertools.product生成组合,然后通过比较将其排除

from itertools import product
a = [1.01, 5.84, 13.86]
b = [6.42,0,18.24]
c = [(1.01,18.24), (13.86,0), (5.84,0)]

combs = list(product(a,b))

[(1.01, 6.42),
 (1.01, 0),
 (1.01, 18.24),
 (5.84, 6.42),
 (5.84, 0),
 (5.84, 18.24),
 (13.86, 6.42),
 (13.86, 0),
 (13.86, 18.24)]


new_combs = [x for x in combs if x not in c]

[(1.01, 6.42),
 (1.01, 0),
 (5.84, 6.42),
 (5.84, 18.24),
 (13.86, 6.42),
 (13.86, 18.24)]

注意:如注释中所述,list周围的combs用于演示,因此您可以查看所有组合。相反,您可能想这样做:

new_combs = [x for x in product(a,b) if x not in c]

更多有关元组和列表比较的信息: How does tuple comparison work in Python?

按位置比较图元组:将第一个元组的第一项与第二个元组的第一项进行比较;如果它们不相等,则为比较的结果,否则考虑第二项,然后考虑第三项,依此类推。

也:
序列类型也支持比较。特别是,通过比较相应的元素按字典顺序比较了元组和列表。这意味着要进行相等比较,每个元素必须进行相等比较,并且两个序列必须具有相同的类型并且具有相同的长度。