[Python]Numpy配列(Array)の累乗和を取得するには？

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

Numpy配列の累乗和を取得する方法を紹介します。

方法

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

sum()メソッド

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

まず、左辺にNumpy配列、右辺に乗数を指定した「**」で演算します。

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

result = (array ** power).sum()

上記のsum()メソッドは、「**」の左辺に指定したNumpy配列の累乗和を返します。

使用例

import numpy as np

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

result = (nums ** 3).sum()

print(result) #2025

numpy.sum()

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

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

そして、numpy.sum()の引数に、左辺にNumpy配列、右辺に乗数を記述した「**」を指定します。

result = np.sum(array ** power)

上記のnumpy.sum()は、「**」の左辺に記述したNumpy配列の累乗和を返します。

使用例

import numpy as np

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

result = np.sum(nums ** 2)

print(result) #285

まとめ

Numpy配列の累乗和を取得する方法は、次の２つです。

• sum()メソッドを使う方法
• numpy.sum()を使う方法