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

標籤：變換   des   href   generate   選擇性參數   lob   目標   ret   表示   

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])

numpy英文解釋文檔

 

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