Python学习：np.nditer( ) 函数的用法

# 迭代器np.nditer提供了一种可以访问一个或者多个数组元素的方法，可以遍历数组。

1，逐个访问元素：

import numpy as np

arr1 = np.arange(9).reshape(3, 3)
print(arr1)
print("---------")

# 按行访问
for i in np.nditer(arr1, order='C'):
    print(i)
print("---------")

# 按列访问
for i in np.nditer(arr1, order="F"):
    print(i)
print("---------")

结果：

[[0 1 2]
 [3 4 5]
 [6 7 8]]
---------
0
1
2
3
4
5
6
7
8
---------
0
3
6
1
4
7
2
5
8
---------