如何在给定轴上迭代numpy数组？

在NumPy中，我们可以使用nditer函数来在给定轴上迭代数组。nditer函数可用于在任意形状的数组上进行迭代，同时提供对每个元素的访问和操作。

以下是在给定轴上迭代NumPy数组的示例代码：

import numpy as np

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

# 在第一个轴上进行迭代
for row in np.nditer(arr, axis=0):
    print(row)
    
# 在第二个轴上进行迭代
for column in np.nditer(arr, axis=1):
    print(column)

输出结果为：

[1 2 3]
[4 5 6]
[7 8 9]
1
2
3
4
5
6
7
8
9

在上述示例中，我们首先创建了一个二维数组arr。然后，使用np.nditer函数和axis参数来指定要在哪个轴上进行迭代。通过使用for循环，我们可以逐个访问数组中的元素。

在第一个for循环中，我们在第一个轴（行）上进行迭代。输出结果显示了每一行的元素。

在第二个for循环中，我们在第二个轴（列）上进行迭代。输出结果显示了每一列的元素。

NumPy的nditer函数提供了更多灵活性，比如可以同时迭代多个数组。您可以参考NumPy官方文档以获取更多关于nditer函数的详细信息：nditer - NumPy官方文档