从 Numpy 3d 数组中保留切片的尺寸

Question

我有一个 3d 数组 a，形状为 a.shape = (10, 10, 10)

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

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

我想保留维度的数量但也要确保被挤压的维度是显示 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。使用整数数组索引时，结果形状始终为"identical to the (broadcast) indexing array shapes"。由于[5]（作为一个数组）的形状为(1,)， a[:,:,[5]] 最终具有 (10,10,1) 的形状。