numpy.apply_along_axis(func, axis, arr, *args, **kwargs): 必選參數:func,axis,arr。其中func是我們自定義的一個函數,函數func(arr)中的arr是一個數組,函數的主要功能就是對數組裡的每一個元素進行變換,得到目標的結果。 ...
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英文解釋文檔