NumPy - nditer迭代器

import numpy as np
a = np.arange(0,8).reshape((2,2,2))
print(a)

输出：

[[[0 1]
  [2 3]]

 [[4 5]
  [6 7]]]

 

使用多维数组迭代器nditer遍历输出每个元素的index和value，

it = np.nditer(a, flags=['multi_index'], order='C')
while not it.finished:
    print('Index:%s, Value:%d' % (it.multi_index, a[it.multi_index]))
    it.iternext()

输出：

Index:(0, 0, 0), Value:0
Index:(0, 0, 1), Value:1
Index:(0, 1, 0), Value:2
Index:(0, 1, 1), Value:3
Index:(1, 0, 0), Value:4
Index:(1, 0, 1), Value:5
Index:(1, 1, 0), Value:6
Index:(1, 1, 1), Value:7