生成未知数量变量的所有组合

Question

我有一个变量列表，其中的数量可以改变。每个变量都有一个下限、上限和一个增量值。例如：

variables = {
   "a": [1, 10, 1],
   "b": [50, 200, 5],
   "c": [50, 300, 10]
}

其中键“a”，1为下限，10为上限，1为增量。

"a" would go from 1 through 10, incrementing by 1
"b" would go from 50 through 200, incrementing by 5
"c" would go from 50 through 300, incrementing by 10
... there can be more or less such keys.

我需要创建 a、b 和 c 的所有可能组合的列表，而无需硬编码嵌套循环，因为变量/键的数量未知，而且我无法理解它。我正在使用 Python 3.7

理想情况下，输出是每个变量的组合表，可能是逗号分隔的值。例如。

a  b  c
x  y  z
x  y  z
x  y  z
x  y  z

但是，只要我可以将输出整理成可以将每个组合作为一组访问的格式，任何事情都可以。比如元组列表。

[
  (x, y, z)
  (x, y, z)
]
...

Answer 1

这里，假设：

variables = {
   "a": [1, 10, 1],
   "b": [50, 200, 5],
   "c": [50, 300, 10]
}

然后你可以使用itertools创建一个迭代器：

from itertools import product, starmap
cartesian_product = product(*starmap(range, variables.values()))

或者，只是为了向您展示星图在做什么：

cartesian_product = product(*(range(*v) for v in variables.values()))

Answer 2

试试：

import numpy as np
lst = [np.arange(v[0], v[1]+1, v[2]) for k,v in variables.items()]
np.array(np.meshgrid(*lst)).T.reshape(-1,len(lst))

Answer 3

试试这个方法：

var=[]
for k in variables.keys():
  var.append(np.arange(variables[k][0],variables[k][1]+1,variables[k][2]))

np.array(np.meshgrid(var[0],var[1],var[2])).T.reshape(-1,3)

对于组合我从here获得了帮助

Answer 4

你可以使用itertools:

import numpy as np
import itertools


a = np.arange(1, 10, 1)
b = np.arange(50, 200, 5)
c = np.arange(50,300,10)

results = itertools.product(a, b, c)
for item in results:
    print(item)

输出：

(1, 50, 50)
(1, 50, 60)
...
...
(9, 195, 270)
(9, 195, 280)
(9, 195, 290)