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

Question

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

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

到目前为止我能想到的唯一方法如下。我使用 product 函数创建了一个元组列表

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 returns 我想要的。然而，我的问题是，如果我想扩展方程式或两个列表，那么打开元组并标记所有内容显然会变成一场噩梦，即使是上面的内容也让我头疼。

有什么更巧妙的方法吗？也许我应该知道任何模拟包？

提前致谢

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)

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

Seeking generalized way to solve equation using all possible combinations of values of two lists in python

python

iteration

simulation

for-loop

tuples