[Python]Numpy配列(Array)の行ごとの積を取得するには？

どうも、ちょげ(@chogetarou)です。

Numpy配列(Array)の行ごとの積を取得する方法を紹介します。

方法

Numpy配列(Array)の行ごとの積を取得するには、numpyのprod()を使います。

まず、numpyからprod()を呼び出します。

そして、prod()の第１引数にnumpy配列、引数「axis」に「１」を指定します。

result = numpy.prod(array, axis=1)

上記のnumpy.prod()は、第１引数に指定したNumpy配列の行ごとの積をまとめた配列を返します。

使用例

import numpy as np

nums = np.array([[1, 2, 3],  
                [4, 5, 6], 
                [7, 8, 9],] )

result = np.prod(nums, axis=1)

print(result) #[  6 120 504]