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

只是將您的索引數組轉換爲元組