Pythonでのnumpy.apply_along_axis()関数の使用



Usage Numpy Apply_along_axis Function Python



関数プロトタイプ
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) #Result: array([12, 16, 20, 24]) #(1+5)*2=12 (2+6)*2=16 and so on np.apply_along_axis(f,1,b) #Result: array([ 6, 22, 38]) #(1+2)*2=6 (5+6)*2=22 and so on