保留一個切片的尺寸從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)。