保留一个切片的尺寸从numpy的3D阵列

Question

我有一个三维阵列，a，形状的说a.shape = (10, 10, 10)

切片时，将尺寸为squeezed自动即

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

I” d要保留尺寸的数量，但也要确保被挤压的尺寸是显示1的尺寸，即

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

我曾经想过重新铸造阵列和传递ndmin，但只是增加了额外的尺寸形状元组的开始，无论片从阵列中a原来的地方。

Answer 1

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

---

a[:,:,5]是basic slicing一个例子。

a[:,:,[5]]是integer array indexing的一个例子 - combined with basic slicing。使用整数数组索引时，生成的形状始终为​​。由于[5]（作为阵列）具有形状(1,)， a[:,:,[5]]最终形状(10,10,1)。