保留Numpy 3D数组中切片的维度

Question

8

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

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

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

我想保留维数的数量，但也确保被压缩的维度是显示1的维度，即

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

我想过重新转换数组并传递ndmin，但这只是将额外的维度添加到形状元组的开头，而不管切片来自数组a的哪个位置。

- Brendan

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- unutbu · Accepted Answer

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

a[:,:,5] 是基本切片的一个例子。

a[:,:,[5]] 是整数数组索引的一个例子 -- 与基本切片相结合。当使用整数数组索引时，结果形状总是与广播索引数组形状相同。由于 [5]（作为一个数组）的形状为 (1,)， a[:,:,[5]] 最终形状为 (10,10,1)。