python中numpy.apply_along_axis()函数的用法

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依次类推