python - 将索引的平面列表创建到数组中的正确正确的 numpy 方法

Question

好吧，这是 super 基本的，不幸的是，这意味着搜索它会得到无数次点击，所有这些都做了不同的和/或更复杂的事情。

考虑这段代码:

shape = (10, 20)
indices = []
for i in range(shape[0]):
    for j in range(shape[1]):
        indices.append([i, j])

或者 indices = itertools.product(range(10), range(20)).

现在，我觉得必须是一个简单的 numpy 函数来做同样的事情？有点像

indices = np.indices_into_shape((10, 20))

我可以通过搜索找到的大多数索引生成函数都会生成多个数组，例如 meshgrid 或 ix_。

Answer 1

您可以堆叠网格:

np.dstack(np.meshgrid(np.arange(10), np.arange(20), indexing='ij')).reshape(-1, 2)