[Python]Numpy配列の横の列(行)の合計値を取得するには？

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

Numpy配列の横の列（行）の合計値を取得する方法を紹介します。

方法

Numpy配列の横の列の合計値を取得するには、sumメソッドを使います。

まず、Numpy配列からsumメソッドを呼び出します。

そして、sumメソッドの引数「axis」に「１」を指定します。

array.sum(axis=1)

上記のsumメソッドは、横の列（行）ごとの合計値を要素にしたNumpy配列を返します。

使用例

import numpy as np

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

rows_sum = nums.sum(axis=1)

print(rows_sum) #[ 6 15 24]