1. 程式人生 > >python中numpy.apply_along_axis()函式的用法

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