我有一个形状为Numpy的数组[1000, 1000, 1000, 3]，大小为3，是最后一个维度，包含3D空间矢量分量的三元组。我如何使用nditer来遍历每个三元组？像这样：
for vec in np.nditer(my_array, op_flags=['writeonly', <???>]):
vec = np.array(something)
我之前已经解决了这个问题，但这是一个简短的示例：
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]
ndindexmulti-index围绕您提供的大小（此处为(2,2,2)）的虚拟数组构造一个迭代器，并将其与next方法一起返回。
因此，您可以ndindex按原样使用，也可以将其用作构建on的模型nditer。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句