[Python]forループでArray.arrayの二乗和を取得するには？

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

forループを使ってArray.arrayの二乗和を取得する方法を紹介します。

方法

forループを使ってArray.arrayの二乗和を取得するには、sum()を使います。

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

sum()の引数に[]を指定します。

そして、[]内でArray.arrauをforループします。

ループ処理で、Array.arrayの要素を2乗します。

result = sum([i**2 for i in array])

上記のsum()は、forループしたArray.arrayの二乗和を返します。

使用例

import array

numbers = array.array('i', [1, 2, 3, 4, 5])

result = sum([i**2 for i in numbers])
    
print(result) #55