2017-10-10 63 views
0

我想liketo知道,如果有一種方法來使用數組從另一個數組中選擇一個元素?如何在python中使用另一個數組選擇多維數組中的元素?

G=np.array([[[0,0,0],[0,0,0],[0,0,0]], 
      [[1,1,1],[1,8,1],[1,1,1]], 
      [[2,2,2],[2,2,2],[2,2,2]]]) 

比方說,我想選擇 '8',我通常只是做

In [101]: G[1,1,1] 
Out[101]: 8 

我打算做IST像

In [102]: c=np.array([1,1,1]) 
In [103]: G[c] 
Out[103]: 8 

我所知道的,這是很容易,它不起作用。 IST有辦法,那確實

回答

0

只需使用tuple()

G[tuple(c)] 
+0

謝謝!我一直在努力幾天,但我沒有找到這個。 – Impeesa

+0

我很高興能夠幫助你!也許你可以接受答案,如果你發現它有幫助嗎? – kaidokuuppa

相關問題