如何迭代Numpy数组的初始尺寸？

如何迭代Numpy数组的初始尺寸？

我之前已经解决了这个问题，但这是一个简短的示例：

vec=np.arange(2*2*2*3).reshape(2,2,2,3)it=np.ndindex(2,2,2)for i in it:    print(vec[i])

生产：

[0 1 2][3 4 5][6 7 8][ 9 10 11][12 13 14][15 16 17][18 19 20][21 22 23]

ndindex``multi-index

围绕您提供的大小（此处为

(2,2,2)

）的虚拟数组构造一个迭代器，并将其与

next

方法一起返回。

因此，您可以

ndindex

按原样使用，也可以将其用作构建on的模型

nditer

。