用矩阵对 numpy 向量进行排序

Question

我已经写了这段代码：

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

其中w 是MxN 矩阵，y 是Nx1 向量。 如果np.shape(y) 返回的形状为 (3,)，则一切正常。但是，当我尝试输入 shape(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,)