排序與基質

Question

numpy的矢量我寫這段代碼：

w = Dist.argsort(kind='mergesort'); 
return y[w]; 

其中w是m×n矩陣，並且y是NX1向量。 如果np.shape(y)返回的形狀是（3，），則一切正常。但是，當我嘗試輸入形狀矢量（3,1）時，我的函數返回一個3維MXNx1矩陣。

有沒有辦法將（N，1）向量減少爲（N，）大小？或者爲（N，1）個矢量規範函數的輸出？

Answer 1

可以與flatten減小（N，1）numpy的陣列（N）：

>>>np.zeros((3, 1)).shape 
(3, 1) 

>>>np.zeros((3, 1)).flatten().shape 
(3,)