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


numpy.apply_along_axis(funcaxisarr*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英文解釋文檔

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM