numpy 遍历二维数组

在numpy中，可以使用循环来遍历二维数组。以下是一个示例代码：

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还提供了一些函数可以用来遍历二维数组，例如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

需要注意的是，使用numpy.nditer()函数遍历二维数组时，会按照内存存储的顺序进行遍历，而不是按照行和列的顺序。如果需要按照行和列的顺序遍历，可以使用order参数来指定遍历的顺序。例如，使用order='F'来按照列的顺序遍历。