numpy配列から複数のindexから要素を取得する方法

Question

numpy配列から複数のindexから要素を取得する方法

import numpy as np
a_2d = np.arange(25).reshape(5, 5)
print(a_2d)

[[ 0  1  2  3  4]
 [ 5  6  7  8  9]
 [10 11 12 13 14]
 [15 16 17 18 19]
 [20 21 22 23 24]]

と適当な配列を定義します。
そこで[0,3]に存在する配列を取得したい場合、

a_2d[0, 3]

3

と出力することができます。
それで私は[0,3]と[0,4]の配列を同時に出力させたいとします。

a_2d[[0,3],[0,4]]

適当に上記のようなコードを打ったところ、

array([ 0, 19])

以下のような配列が返って気ました。

これは0行0列と3行4列にあたる2つの数字であることが分かりました。
そこで[0,3]と[0,4]の配列を同時に出力させるような方法。

array([ 3, 4])

上記のような結果が返ってくるような方法はあるでしょうか？
分かる方がいれば教えていただけると幸いです。

Answer 1

ちょっと惜しいですね。

>>> import numpy as np
>>> a_2d = np.arange(25).reshape(5, 5)
>>> a_2d[[0, 0], [3, 4]]
array([3, 4])

説明。カンマで区切ったものが各axisに対応する訳です。

a_2d[  # ここがnumpy配列に対する添字表記のはじまり）
     [0, 0],  # axis=0（行）に対する指定 
     [3, 4]   # axis=1（列）に対する指定 
]  # 添字表記終わり

Comments

[yu__]

最後にその発想に至らなかったのが恥ずかしいです。
大変勉強になりました！ありがとうございました！！