Numeración de dimensiones arbitrarias

Question

Me gustaría dividir una matriz numpy para obtener el índice i-ésimo en la última dimensión. Para una matriz 3D, esto sería:

slice = myarray[:,:,i] 

Pero estoy escribiendo una función donde puedo tomar una serie de medidas arbitrarias, por lo que para una matriz de 4D que necesitaría myarray[:,:,:,i], y así sucesivamente. ¿Hay alguna forma de que pueda obtener esta porción para cualquier matriz sin tener que escribir explícitamente las dimensiones de la matriz?

Answer 1

Hay ... o Ellipsis, que hace exactamente esto:

slice = myarray[...,i] 

Elipsis es el objeto pitón, si debe querer usarlo fuera de la notación de corchetes.

Answer 2

En realidad, acabo de encontrar la respuesta. Como se indica en numpy's documentation, esto se puede hacer con el objeto slice. En mi caso particular, esto se hace:

idx = [slice(None)] * (myarray.ndim - 1) + [i] 
my_slice = myarray[idx] 

El slice(None) es equivalente a elegir todos los elementos de ese índice, y el último [i] selecciona un índice específico para la última dimensión.

Answer 3

En términos de rebanar una dimensión arbitraria, las anteriores excelentes respuestas se puede extender a:

indx = [Ellipsis]*myarray.ndim 
indx[slice_dim] = i 
slice = myarray[indx] 

Esto devuelve el rebanada de cualquier dimensión slice_dim - slice_dim = -1 reproduce las respuestas anteriores. Para completar - las dos primeras líneas de la lista anterior se pueden condensar a:

indx = [Ellipsis]*(slice_dim) + [i] + [Ellipsis]*(myarray.ndim-slice_dim-1) 

aunque encuentro la versión anterior más legible.