numpy.ndarrayからindexを指定して要素を抽出したい

Question

実現したいこと

numpyのndarrayから、インデックスを格納した配列を使ってndarrayの要素を抽出したいです。
下記のコードのコメントアウトに書いてあるようなことを想定しています。
for文での実装はできたのですが、numpyでスッキリ記述する方法を見つけることができず、教えていただけますと幸いです。
宜しくお願い致します。

hoge = np.array([[0, 1, 2], [3, 4, 5], [10, 11, 12]])
idx = np.array([1, 1, 0])

"""
idxを使った処理で
hoge[0, idx[0]]の1,
hoge[1, idx[1]]の4,
hoge[2, idx[2]]の10
を抽出して[1, 4, 10]を出力したい

for文で実装すると
out = []
for l, i in zip(hoge, idx):
    out.append(l[i])
"""

Answer 1

ファンシーインデックスを使います。

>>> hoge = np.array([[0, 1, 2], [3, 4, 5], [10, 11, 12]])
>>> idx0 = np.arange(3)
>>> idx1 = np.array([1, 1, 0])
>>> print(hoge[idx0,idx1])
[ 1  4 10]

Comments

[egnoi]

ありがとうございます！