在python中按行对数组进行排序

Question

我知道按列对 numpy 数组 arr 进行排序（仅针对特定列，例如其第二列）可以通过以下方式完成：

arr[arr[:,1].argsort()]

我如何理解此代码示例的工作原理：argsort 对arr 的第二列的值进行排序，并将相应的索引作为数组提供。该数组作为行号提供给arr。我的解释正确吗？

现在我想知道如果我想根据第二行而不是第二列对数组arr 进行排序怎么办？是在排序之前转置数组并在排序后将其转回的最简单方法，还是有一种方法可以像以前一样（通过给出一个包含我们希望显示的列数的数组）？

我没有使用(n,n)array[(n,)array]（n 是二维数组的大小），而是尝试使用(n,n)array[(n,1)array] 之类的方法来指示列数，但它不起作用。

我想要的示例：

arr = [[11,25],[33,4]] => 基本数组

arr_col2=[[33,4],[11,25]] => 我用argsort() 得到的数组

arr_row2=[[25,11],[4,33]] => 数组我尝试使用argsort() 以简单的方式获得但没有成功

Answer 1

我假设 arr 是一个 numpy 数组？我在 python 的任何其他上下文中都没有看到语法arr[:,1]。在您的问题中值得一提！

假设是这种情况，那么您应该使用

arr.sort(axis=0)

按列排序

arr.sort(axis=1)

按行排序。 （两者都就地排序，即改变arr的值。如果你不想要这个，你可以先将arr复制到另一个变量中，然后应用sort。）

如果您只想对一行进行排序（在本例中为第二行），那么

arr[1,:].sort()

有效。

编辑：我现在了解您要解决的问题。您想对矩阵中的列重新排序，以便第 n^th 行按升序排列。你可以简单地做到这一点

arr[:,arr[1,:].argsort()]

（这里我们按第二行排序）。