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

  • A+
所属分类:python数据分析

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

Python

numpy.apply_along_axis(func, axis, arr, *args, **kwargs):

必选参数:func,axis,arr。其中func是我们自定义的一个函数,函数func(arr)中的arr是一个数组,函数的主要功能就是对数组里的每一个元素进行变换,得到目标的结果。

其中axis表示函数func对数组arr作用的轴。

可选参数:*args, **kwargs。都是func()函数额外的参数。

返回值:numpy.apply_along_axis()函数返回的是一个根据func()函数以及维度axis运算后得到的的数组.

实例:

def my_func(a):

  return (a[0] + a[-1]) * 0.5

b=np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12]])

np.apply_along_axis(my_func, 0, b)

输出: array([ 5., 6., 7., 8.])

np.apply_along_axis(my_func, 1, b)

输出: array([ 2.5, 6.5, 10.5])

weinxin
我的微信公众号
爱真理,得永生!          爱在灵灵久博客,网罗天下,福利大家!

发表评论

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen: