python - 在python中使用两个列表的值的所有可能组合寻求求解方程的通用方法

Question

我想看看每个列表的乘积的哪些组合可以解方程，“j=k-l+m”

stake=[0,2,5,10]
odds=[1,2,5,10]

到目前为止，我能想到的唯一方法如下。我使用像这样的产品功能创建了一个元组列表

from itertools import product

stake_odds=(list(product(stake,odds)))
print(stake_odds)
>>>[(0, 1), (0, 2), (0, 5), (0, 10), (2, 1), (2, 2), (2, 5), (2, 10), (5, 1), (5, 2), (5, 5), (5, 10), (10, 1), (10, 2), (10, 5), (10, 10)]

然后我创建了一个进一步的列表，其中包含长度为等式中变量数量的元组组合:

stake_odds2=(list(product(stake_odds, repeat=3)))

然后我执行以下操作

pos_combos=[]
for i in stake_odds2:
    x,y,z=i
    a,b=x
    c,d=y
    e,f=z
    k=a*b
    l=c*d
    m=e*f
    j=k-l+m
    if j > 0:
        pos_combos.append(i)

打印 pos_combos 返回我想要的。然而，我的问题是，如果我想扩展方程或两个列表，那么解压缩元组并标记所有内容显然变成了一场噩梦，即使上面的内容也让我头疼。
有没有更巧妙的方法来做到这一点？也许我应该知道任何模拟包？
提前致谢

Answer 1

无需解压所有这些变量，只需取元组的乘积然后进行最后的数学运算即可。

from itertools import product
import numpy as np
stake=[0,2,5,10]
odds=[1,2,5,10]
stake_odds=(list(product(stake,odds)))
stake_odds2=(list(product(stake_odds, repeat=3)))

for i in stake_odds2:
    t = [np.prod(x) for x in i]
    if t[0]-t[1]+t[2]>0:
        pos_combos.append(i)