按照第一行对numpy 2d数组进行排序，保持列不变。

Question

在Python中，我有一个形如以下的numpy数组：

[4 8 2 0 5]
[3 1 6 8 1]
[2 2 6 0 3]
[9 7 6 7 8]
[5 8 1 1 4]

我希望按照从左到右第一行的值进行升序排序，同时保持列的完整性。实际数组的维度未指定且非常巨大，因此使用for循环编写的代码过于缓慢。结果应该是：

[0 2 4 5 8]
[8 6 3 1 1]
[0 6 2 3 2]
[7 6 9 8 7]
[1 1 5 4 8]

我可以使用argsort获取按正确顺序排序的列索引的行向量，但不知道如何实际构建新数组。

- Joe Gallagher

---

可能是 sorting arrays in numpy by row 的重复问题。 - Mr. T

Answer 1

源数组：

In [215]: a
Out[215]:
array([[4, 8, 2, 0, 5],
       [3, 1, 6, 8, 1],
       [2, 2, 6, 0, 3],
       [9, 7, 6, 7, 8],
       [5, 8, 1, 1, 4]], dtype=int64)

使用Numpy索引：

In [218]: a[:, a[0].argsort()]
Out[218]:
array([[0, 2, 4, 5, 8],
       [8, 6, 3, 1, 1],
       [0, 6, 2, 3, 2],
       [7, 6, 9, 8, 7],
       [1, 1, 5, 4, 8]], dtype=int64)

使用Pandas:

In [212]: pd.DataFrame(a).sort_values(0, axis=1).values
Out[212]:
array([[0, 2, 4, 5, 8],
       [8, 6, 3, 1, 1],
       [0, 6, 2, 3, 2],
       [7, 6, 9, 8, 7],
       [1, 1, 5, 4, 8]], dtype=int64)

- MaxU - stand with Ukraine

---

看起来可以正常工作，但不幸的是，这是作为一个任务的一部分，不允许使用其他库。 - Joe Gallagher

@JoeGallagher，你不能使用NumPy对NumPy数组进行排序吗？ - MaxU - stand with Ukraine

不行，不能使用pandas，我的回复是在你编辑之前做出的。Numpy的解决方案可行。 - Joe Gallagher