どうも、ちょげ(@chogetarou)です。
Numpy配列の行ごとの累乗和を取得する方法を紹介します。
方法
Numpy配列(Array)の行ごとの累乗和を取得する方法は、２つあります。
sum()メソッド
１つは、sum()メソッドを使う方法です。
まず、左辺にNumpy配列、右辺に乗数を指定した「**」で演算します。
そして、「**」の演算結果からsum()メソッドを呼び出します。
sum()メソッドの引数「axis」に「1」を指定します。
result = (array ** power).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]numpy.sum()
もう１つは、numpy.sum()を使う方法です。
まず、numpyからsum()を呼び出します。
そして、numpy.sum()の第１引数に、左辺にNumpy配列、右辺に乗数を記述した「**」を指定します。
numpy.sum()の引数「axis」に「1」を指定します。
result = np.sum(array ** power, 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]まとめ
Numpy配列の行ごとの累乗和を取得する方法は、次の２つです。
- sum()メソッドを使う方法
- numpy.sum()を使う方法
コメント