Python 用矩阵排序numpy向量

我已经编写了这段代码：

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

其中，

w

是MxN矩阵，

y

是Nx1向量。 如果

np.shape（y）

返回的shape为（3，），则一切正常。但是，当我尝试输入形状向量（3,1）时，我的函数返回一个三维MXNx1矩阵

有没有办法把（N，1）向量缩小到（N，）大小？或者要规范化（N，1）向量的函数输出？

您可以使用

展平将（N，1）numpy数组减少为（N，）：

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

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