Numpy配列の特定の要素の和のみを求めたい

Question

Numpy配列のうち、特定の要素、たとえば1番目と3番目だけの和を求めたい、という場合の方法が分からずにおります。以下の様にしますと、エラーになってしまいます。

import numpy as np
x = np.array([1,2,3,4])
np.sum(x[1,3])

エラー：IndexError Traceback (most recent call last)
<ipython-input-38-1f75dc87a566> in <module>
1 x = np.array([1,2,3,4])
----> 2 np.sum(x[1,3])

IndexError: too many indices for array

Web上で様々調べてきたのですが、分からずにおります。。方法がありましたら、ご教示下さい。

Answer 1

指定したインデックスの値だけを取得する (fancy indexing) 場合は、以下のようにリストで指定します。

import numpy as np

x = np.array([1, 2, 3, 4])
print(np.sum(x[[1, 3]]))