np.apply_along_axis 的作用是在数组的每一个轴上执行函数,它的一个常见的例子是对于二维数组的每一行或每一列进行计算:
import numpy as np
def calculate_row_sum(row):
return sum(row)
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 计算每一行的和
row_sums = np.apply_along_axis(calculate_row_sum, 1, a)
print(row_sums)
# 输出:[ 6 15 24]
# 计算每一列的和