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

python

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

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

スポンサーリンク

方法

インターフェース, インターネット, プログラム, ブラウザ, Www

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

sum()メソッド

1つは、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()

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

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

そして、numpy.sum()の第1引数に、左辺に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配列の行ごとの累乗和を取得する方法は、次の2つです。

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

コメント

タイトルとURLをコピーしました