python - Numpy索引切片不丢失维度信息

Question

我正在使用 numpy，希望在不丢失维度信息的情况下索引一行。

import numpy as np
X = np.zeros((100,10))
X.shape        # >> (100, 10)
xslice = X[10,:]
xslice.shape   # >> (10,)  

在此示例中，xslice 现在是 1 维，但我希望它是 (1,10)。在 R 中，我会使用 X[10,:,drop=F]。 numpy中有类似的东西吗？我在文档中找不到它，也没有看到提出类似的问题。

谢谢!

Answer 1

另一个解决方案是这样做

X[[10],:]

或

I = array([10])
X[I,:]

当通过索引列表(或数组)执行索引时，会保留数组的维数。这很好，因为它让您可以在保持尺寸和挤压之间进行选择。