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

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

Numpy配列(Array)の縦の列（カラム）ごとの積を取得する方法を紹介します。

方法

Numpy配列(Array)の縦の列（カラム）ごとの積を取得するには、numpyのprod()を使います。

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

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

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

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

使用例

import numpy as np

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

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

print(result) #[ 28  80 162]