python - NumPy 配列のスライス - 特定のインデックスに対応する値を選択する

Question

特定のインデックスに対応する配列から要素を選択する簡単な方法はありますか? 例: インデックスの配列 IDX = array((3,7,5)) と配列 A = array((2,5,3,1,7,5,1,9,7,4, 2,5)) から、IDX で指定されたインデックスに対応する A の要素を選択したいと思います。

つまり、私はこのようなものを探しています:

A[IDX] = array((1,9,5)) or  
A[3,7,5] = array((1,9,5)).

Answer 1

Numpy を使用すると、次のようになります。

>>> import numpy as np
>>> IDX = np.array([3,7,5])
>>> A = np.array([2,5,3,1,7,5,1,9,7,4,2,5])
>>> A[IDX]
array([1, 9, 5])

Answer 2

私は numpy をインストールしていませんが、標準の python ではリスト内包表記を使用できます。numpy 配列の動作は大きく異なるべきではないと思います。

>>> IDX = (3,7,5)
>>> A = (2,5,3,1,7,5,1,9,7,4,2,5)
>>> [A[i] for i in IDX]
[1, 9, 5]