Numpy 二维数组遍历：循环与 nditer() 函数详解

在 Numpy 中，可以使用循环或 numpy.nditer() 函数来遍历二维数组。

循环遍历二维数组

可以使用嵌套循环来遍历二维数组的每个元素。以下是一个示例代码：

import numpy as np

# 创建一个 3x3 的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 循环遍历二维数组
for i in range(arr.shape[0]):
    for j in range(arr.shape[1]):
        print(arr[i][j])

输出结果：

1
2
3
4
5
6
7
8
9

使用 numpy.nditer() 函数遍历二维数组

numpy.nditer() 函数可以方便地遍历数组的元素，但需要注意的是，它按照内存存储的顺序进行遍历，而不是按照行和列的顺序。

以下是一个示例代码：

import numpy as np

# 创建一个 3x3 的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 使用 nditer 函数遍历二维数组
for i in np.nditer(arr):
    print(i)

输出结果：

1
2
3
4
5
6
7
8
9

控制 nditer() 函数的遍历顺序

如果需要按照行和列的顺序遍历，可以使用 order 参数来指定遍历顺序。例如，使用 order='F' 来按照列的顺序遍历。