python - 複数のインデックスを使用して NumPy 配列から値を取得する方法

Question

次のような NumPy 配列があります。

arr = np.array([100.10, 200.42, 4.14, 89.00, 34.55, 1.12])

この配列からインデックスで複数の値を取得するにはどうすればよいですか?

たとえば、インデックス位置 1、4、および 5 の値を取得するにはどうすればよいでしょうか?

私はこのようなことを試みていましたが、これは正しくありません:

arr[1, 4, 5]

Answer 1

このようにしてみてください：

>>> arr = np.array([100.10, 200.42, 4.14, 89.00, 34.55, 1.12])
>>> arr[[1,4,5]]
array([ 200.42,   34.55,    1.12])

多次元配列の場合:

>>> arr = np.arange(9).reshape(3,3)
>>> arr
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> arr[[0, 1, 1], [1, 0, 2]]
array([1, 3, 5])

Answer 2

あなたは近くにいた

>>> print arr[[1,4,5]]
[ 200.42   34.55    1.12]