다차원 numpy 배열의 인덱스로 Numpy 배열

Question

이것은 완전히 간단 해 보입니다. 어떤면에서는 마치 대괄호를 없앨 필요가있는 것처럼 보입니다. 배열을 사용하여 다차원 배열의 인덱스가되고 싶습니다. 나는 총 거짓말이어야한다. 여기 내 최소 예가 있습니다 :

import numpy as np 

## I know this is a dumb way to initialize. 
state = np.array([[[1,2,3],[4,5,6],[7,8,9]],[[10,11,12],[13,14,15],[16,17,18]]]) 
print('state\n',state) 
a = np.array([0,0,0]) 
print('a',a) 

## I want to use a as the indices of state a la: 
print('state[a]\n',state[a]) 

## If I could just get rid of the brackets I'd be fine: 
print('state[0,0,0]\n',state[0,0,0]) 

## This gets me what I want, but it is UGLY in non-minimal case: 
print('state[a[0],a[1],a[2]]\n',state[a[0],a[1],a[2]]) 

고마워요!

Answer 1

그냥 인덱스 배열을 튜플로 변환하십시오.