numpy_nditer函数

numpy的nditer函数用于对数组中的元素进行迭代访问。

import numpy as np

arr = np.array([[0, 1, 2], [3, 4, 5]])
print(arr)
print("----------")
for x in np.nditer(arr):
    print(x, end=" ")

执行结果：

[[0 1 2]
 [3 4 5]]
----------
0 1 2 3 4 5