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

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

lib ast view lin 分享 ocs func 作用 是我

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()函數的用法