브로드 캐스팅을 사용하여 2 차원 인덱스를 가진 2D numpy 배열에서 요소를 얻는 방법?

Question

행 인덱스 열 목록을 사용하여 요소를 추출 할 2D numpy 배열이있는 경우.

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]