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

Question

试图生成一个排除 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)

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

注意：如 cmets 中所述，combs 周围的 list 用于演示，以便您可以查看所有组合。相反，您可能想要这样做：

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

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

元组逐个位置进行比较：第一个元组的第一项与第二个元组的第一项进行比较；如果它们不相等，这是比较的结果，否则考虑第二项，然后考虑第三项，依此类推。

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