切片陣列的形狀

Question

我在切片後有數組計算的問題。問題是由切片陣列形狀的不確定性引起的。

例如，我有一個2D陣列data與(118,3)形狀，然而，當我只用數據的第一列作爲以下，我只能確定形狀(118,)。除非我使用reshape，否則無法確定#column。我不懂爲什麼。

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

結果是：(118, 3) (118,)。

我找到了類似的東西question問在stackoverflow上。但它沒有回答我的困惑。

Answer 1

給出一個維度的具體索引，從而減少結果中的維度。如果你想保持這個維度，你必須提供一個元素片：

print data[:, 0:1].shape 

結果(118, 1)。