Dynamically ein Vektor entlang 1 Dimension in 'n' dimensional numpy Array erhalten

Question

I eine numpy Variable haben, die 'n' Dimensionen, beispielsweise sein kann: game_board = np.zeros((4,3,3), dtype=np.int8)

Ich möchte ein Vektor entlang der ersten Dimension erhalten, basierend auf einem Vektor choose_vector choose_vector = np.array([x,y],dtype=np.int8)

ich weiß, wie ich dies tun kann statisch:

game_board[:, x, y] 
# will return [0,0,0,0], the (x,y)th element from 1st dimension 

aber alles, was ich habe dies versucht, so weit dabei die 012.374 mithat nicht funktioniert:

game_board[:, choose_vector] 
# returns 
[[[0 0 0] 
    [0 0 0]] 

[[0 0 0] 
    [0 0 0]] 

[[0 0 0] 
    [0 0 0]] 

[[0 0 0] 
    [0 0 0]]] 

print(game_board[choose_vector]) 
# returns 
[[0,0,0]] 

, wie ich den Index für game_boardchoose_vector gegeben, um konstruieren Sie das gleiche Ergebnis wie game_board[:, x, y]

ich es jedes dimensionalen Spielbrett erweitern würde dann zu bekommen, aber ich kann es wahrscheinlich funktionieren, wenn ich weiß, wie die oben genannten zu tun :)

Answer 1

Ich habe dies mit Hilfe von FlashTek ausgearbeitet. Statt np.newaxis zu verwenden, scheint mit Scheibe (None) für :

import numpy as np 
x,y = 0,0 

game_board = np.zeros((4,3,3), dtype=np.int8) 
choose_vector = np.array([x, y], dtype=np.uint8) 
game_board[[slice(None)] + choose_vector.tolist()] 

Answer 2

Dies könnte nicht die sauberste Lösung, aber es tut, was Sie wollen:

import numpy as np 
x,y = 0,0 

game_board = np.zeros((4,3,3), dtype=np.int8) 
choose_vector = np.array([x, y], dtype=np.uint8) 
game_board[[np.newaxis] + choose_vector.tolist()] 

Der Trick ist, dass Sie die : in Ihrem statischen Ansatz mit einem np.newaxis innerhalb eines Python list "ersetzen" können.