对numpy数组中的每一行应用函数可以使用numpy的apply_along_axis函数。
apply_along_axis函数的语法如下: numpy.apply_along_axis(func1d, axis, arr, *args, **kwargs)
参数说明:
示例代码如下:
import numpy as np
# 自定义函数,计算每一行的和
def row_sum(row):
return np.sum(row)
# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 对每一行应用row_sum函数
result = np.apply_along_axis(row_sum, 1, arr)
print(result)
输出结果:
[ 6 15 24]
在这个例子中,我们定义了一个自定义函数row_sum,用于计算每一行的和。然后使用apply_along_axis函数对数组arr的每一行应用row_sum函数,得到每一行的和。最后将结果打印出来。
推荐的腾讯云相关产品:腾讯云云服务器(CVM)和腾讯云函数(SCF)。
领取专属 10元无门槛券
手把手带您无忧上云