[Python]Numpy配列の特定の範囲の合計を取得する(ndarray get sum of range)には？

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

Numpy配列(ndarray)の指定した範囲の合計値を取得する方法を紹介します。

方法

Numpy配列(ndarray)の指定した範囲の合計値を取得するには、numpy.sum()とスライスを使います。

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

そして、numpy.sum()の引数に、Numpy配列(ndarray)を範囲でスライスした結果を指定します。

#arr=対象のNumpy配列, start=範囲の最初の位置, end=範囲の最後の位置
np.sum(arr[start:end]) #startからendまでの範囲(endは含まない)の合計値

上記のnumpy.sum()は、スライスで指定した範囲の合計値を返します。

「arr[start:end]」の「end」まで含めたい場合は、「end」に「+1」します。

np.sum(arr[start:end + 1]) #endを含める

import numpy as np

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

result = np.sum(nums[3:7])

print(result)

出力：
22