python - 切片数组的形状

Question

我在切片后的数组计算方面遇到问题。该问题是由切片数组的形状的不确定性造成的。

例如，我有一个形状为 (118,3) 的二维数组 data，但是，当我仅使用第一列数据时，如下所示，我只能确定形状为(118,)。除非我使用reshape，否则无法确定#column。我不懂为什么。

print shape(data), shape(data[:, 0])

结果是:(118, 3) (118,)。

我发现了类似的question在 stackoverflow 上问。但它并没有解答我的困惑。

Answer 1

为某个维度提供具体索引，会减少结果中的该维度。如果你想保留这个维度，你必须提供一个单元素切片:

print data[:, 0:1].shape

结果为(118, 1)。