1.函数原型
numpy.apply_along_axis(func, axis, arr, *args, **kwargs)
2.作用:
将arr数组的每一个元素经过func函数变换形成的一个新数组
3.参数介绍:
其中func,axis,arr是必选的
func是我们写的一个函数
axis表示函数func对arr是作用于行还是列
arr便是我们要进行操作的数组了
可选参数:*args, **kwargs。都是func()函数额外的参数。
4.列子`
import numpy as np
def f(a):
return (a[0]+a[1])*2
b=np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12]])
np.apply_along_axis(f,0,b)
#结果:array([12, 16, 20, 24])
#(1+5)*2=12 (2+6)*2=16依次类推
np.apply_along_axis(f,1,b)
#结果:array([ 6, 22, 38])
#(1+2)*2=6 (5+6)*2=22依次类推
---------------------
作者:sorrythanku
来源:CSDN
原文:https://blog.csdn.net/starmoth/article/details/83832458
版权声明:本文为博主原创文章,转载请附上博文链接!