ブロードキャストを使用してリストを2次元インデックスで持つ2D numpy配列から要素を取得する方法は？

Question

2行目の配列があり、行、colインデックスのリストを使用して要素を抽出したい場合。

xy = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) 
idx = np.array([[0, 0], [1, 1], [2, 2]]) 

ループのためのソリューション：そこIDXはタプルのリストがあるが、何がある場合、私は解決策を望んでいない場合

print(elements) 
>> [1, 5, 9] 

私は解決策を見つけた：出力

elements = list() 
for i in idx: 
    elements.append(xy[idx[i][0], xy[idx[i][1]) 

idxを最初にタプルに変換する必要があります。

Answer 1

idy = zip(*idx) 
    output = xy[idy]