python - 是否有一个 numpy 函数返回给定大小的数组索引矩阵？

Question

我有一个 28x28 的 numpy 数组，我想创建一个新的元组 numpy 数组，每个元组代表原始数组中的一个索引，类似于:[(0, 0), (0, 1), ..., (0, 28), (1, 0), (1, 1),..., (1, 28),...等等。

有什么简单的方法吗？

Answer 1

有np.ndindex ，您可以使用它在给定数组形状的情况下迭代 N 维索引:

a = np.random.rand(28,28)

list(np.ndindex(a.shape))
# [(0, 0), (0, 1), (0, 2), (0, 3), (0, 4), (0, 5), (0, 6), (0, 7), (0,...