如何按一列对 numpy 中的值进行排序 - Python

在使用 numpy 时，我们经常需要对数组进行排序。本文将介绍如何按照某一列对 numpy 数组中的值进行排序。

首先，让我们创建一个示例数组：

import numpy as np

arr = np.array([[3, 2, 1],
                [6, 5, 4],
                [9, 8, 7]])

在上例中，我们创建了一个 3x3 的矩阵，其中包含三行、每行三个数字。

按列排序

如果我们要按列对数组中的值进行排序，可以使用 np.sort() 方法。例如，如果我们要按照每一列的第一行对数组进行排序，可以像下面这样做：

arr_sorted = arr[:, arr[0, :].argsort()]

该行代码将对原数组按照第一行进行排序，排序后的结果存储在 arr_sorted 变量中。这里，我们使用了 argsort() 方法找到排序后的索引，并使用数组切片的方式将整个数组按照索引重新排列。

如果我们要按照每一列的第二行排序，可以使用类似的方法：

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

同理，如果要按第三行排序，可以修改索引的位置：

arr_sorted = arr[:, arr[2, :].argsort()]

结论

本文介绍了如何对 numpy 数组中的值按照一列进行排序。我们使用了 argsort() 方法找到排序后的索引，并使用数组切片重新排列整个数组。这种方法对于处理任何需要按照列排序的数据都很有用，例如，处理 csv 文件或数据库中的数据。