Numpy array multidimensionale affettare

Question

Supponiamo ho definita una matrice 3x3x3 NumPy con

x = numpy.arange(27).reshape((3, 3, 3)) 

Ora, posso ottenere una matrice contenente il (0,1) di ciascun elemento sottomatrice 3x3 con x[:, 0, 1], che restituisce array([ 1, 10, 19]). Cosa succede se ho una tupla (m, n) e voglio recuperare l'elemento (m, n) di ogni sottoarray (0,1) memorizzato in una tupla?

Ad esempio, supponiamo di avere t = (0, 1). Ho provato x[:, t], ma non ha il comportamento corretto - restituisce le righe 0 e 1 di ogni sottoarray. La soluzione più semplice che ho trovato è

x.transpose()[tuple(reversed(t))].transpose() 

ma sono sicuro che ci deve essere un modo migliore. Naturalmente, in questo caso, potrei fare x[:, t[0], t[1]], ma questo non può essere generalizzato al caso in cui non so quante dimensioni hanno x e t.

Answer 1

è possibile creare l'indice di tuple prima:

index = (numpy.s_[:],)+t 
x[index] 

Answer 2

soluzione HYRY è corretto, ma ho sempre trovato r_, c_ e s_ trucchi indice di NumPy ad essere un po 'strano alla ricerca. Così qui è la cosa equivalente utilizzando un oggetto slice:

x[(slice(None),) + t] 

Questo singolo argomento per tagliare è la posizione di arresto (cioè None cioè tutti nello stesso modo in cui x[:] è equivalente a x[None:None])