长话短说，我正在尝试生成一组 numpy 数组的所有可能排列。我有三个数字 [j,k,m]，我想为每个数字 [J,K,M] 指定一个最大值。那么我将如何获得这些值下的所有数组组合？我怎么能强制 k 值总是一样好？例如：
因此，将最大值设置为 [1,2,2]，排列将是：[0,0,0]、[0,0,1]、[0,0,2]、[0, 2,0], [0,2,1], [0,2,2], [1,0,0], [1,0,1] ...
我意识到我没有任何代码可以展示，但恐怕我真的不知道从哪里开始。
从其他答案看来 sympy 会有用吗？
【问题讨论】：
我找到了您可能感兴趣的答案here 并对其进行了概括。因此，您可以像这样为每个项目构建可能值的列表：
X = [[0, 1], [0, 1, 2], [0, 1, 2]]
然后使用：
np.array(np.meshgrid(*X)).T.reshape(-1, len(X))
输出包含您想要的 18 个项目。实际上，如果你只有最大值[J, K, L]，你可以使用X = [range(J+1), range(K+1), range(L+1)]构造X
【讨论】：
def permute(j, k, m): a = np.stack([[i, 2*l, n] for i in range(j + 1) for l in range(k + 1) for n in range(m + 1)]) return a