[Python]Numpy配列(Array)の行ごとの二乗和を取得する(get rows sum of squares)には？

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

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

方法
1. numpy.sum()
2. sum()メソッド
まとめ

方法

Numpy配列(Array)の行ごとの二乗和を取得する方法は、２つあります。

numpy.sum()

１つは、numpy.sum()を使う方法です。

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

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

result = np.sum(array ** 2, axis=1)

上記のnumpy.sum()は、「**」で演算したnumpy配列の行ごとの二乗和を返します。

使用例

import numpy as np

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

result = np.sum(nums ** 2, axis=1)

print(result) #[ 14  77 194]

sum()メソッド

もう１つは、sum()メソッドを使う方法です。

まず、左辺にnumpy配列、右辺に「２」を記述した「**」で演算します。

そして、「**」の演算結果からsum()メソッドを呼び出します。

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

result = (array ** 2).sum(axis=1)

上記のsum()メソッドは、「**」で演算したnumpy配列の行ごとの二乗和を返します。

使用例

import numpy as np

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

result = (nums ** 2).sum(axis=1)

print(result) #[ 14  77 194]