Preservar las dimensiones de una rebanada de un Numpy matriz 3d

Question

tengo una matriz 3d, a, de forma decir a.shape = (10, 10, 10)

Cuando rebanar, las dimensiones son squeezed automáticamente es decir

a[:,:,5].shape = (10, 10)

I' d desea conservar el número de dimensiones pero también asegúrese de que la dimensión que se apretó es la que muestra 1 es decir

a[:,:,5].shape = (10, 10, 1)

he pensado en volver a colar la matriz y pasar ndmin pero que aumenta las dimensiones adicionales del inicio de la tupla forma, independientemente de donde vino de la rodaja de la matriz a.

Answer 1

a[:,:,[5]].shape 
# (10,10,1) 

---

a[:,:,5] es un ejemplo de basic slicing.

a[:,:,[5]] es un ejemplo de integer array indexing - combined with basic slicing. Cuando se utiliza la indexación de matriz de enteros, la forma resultante siempre es "identical to the (broadcast) indexing array shapes". Desde [5] (como una matriz) tiene forma (1,), a[:,:,[5]] termina teniendo forma (10,10,1).