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

Question

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

切片时，维度会自动压缩，即

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)。