1. ホーム
  2. python

[解決済み] Numpy行列の行を反復して、それぞれ関数を適用する?

2022-03-08 12:08:42

質問

行列を反復処理し、各行に関数を適用できるようにしたいです。Numpyの行列でそれを行うにはどうすればよいですか?

どのように解決するのですか?

を使用することができます。 numpy.apply_along_axis() . 配列が2Dであると仮定すると、次のように使うことができます。

import numpy as np
mymatrix = np.matrix([[11,12,13],
                      [21,22,23],
                      [31,32,33]])
def myfunction(x):
    return sum(x)

print(np.apply_along_axis(myfunction, axis=1, arr=mymatrix))
#[36 66 96]