在python中生成多维网格

Question

我有一个离散的8维有界空间，我希望在（N，8）形状的所有可能组合上得到网格。它应该看起来像：

import numpy as np
myGrid = np.array([[1,1,1,1,1,1,1,1],[1,1,1,1,1,1,1,2],...])

所有8个维度的界限都不相等。

Answer 1

您可以使用indices和moveaxis：

np.moveaxis(np.indices(<your shape>), 0, -1).reshape(-1, 8)

这将从零开始，因此添加1以获得您想要的输出。