6、NumPy数组操作全解析

NumPy数组操作全解析

1. 数组迭代

在Python中，迭代数组中的元素非常简单，只需使用 for 循环结构即可。以下是一个一维数组迭代的示例：

for i in a:
    print(i)

当处理二维数组时，可以使用嵌套的 for 循环。第一个循环扫描数组的行，第二个循环扫描列。例如：

for row in A:
    print(row)

若要逐个元素进行迭代，可以使用 A.flat 配合 for 循环：

for item in A.flat:
    print(item)

不过，NumPy提供了比 for 循环更优雅的解决方案—— apply_along_axis() 函数。该函数接受三个参数：聚合函数、应用迭代的轴以及数组。当 axis 为0时，按列评估元素；当 axis 为1时，按行评估元素。示例如下：

import numpy as np
np.apply_along_axis(np.mean, axis=0, arr=A)