用矩阵对numpy向量排序

Question

我写了这段代码：

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

其中w是MxN矩阵，y是Nx1向量。 如果np.shape(y)返回的shape是（3，）的话，一切都正常。但是，当我尝试输入一个形状向量（3,1）时，我的函数返回一个三维MXNx1矩阵。你知道吗

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

Answer 1

可以使用flatten将（N，1）numpy数组缩减为（N，1）：

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

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