如何按特定列对 NumPy 数组进行排序？

按列对 NumPy 数组进行排序：综合指南

在 NumPy 中按特定列对数组进行排序是数据操作的一个重要方面。本指南提供了如何执行此操作的全面说明，解决了以下问题：

如何按 NumPy 数组的第 n 列对它进行排序？

例如，给定：

a = np.array([[9, 2, 3],
           [4, 5, 6],
           [7, 0, 5]])

您想要对行进行排序的第二列得到：

array([[7, 0, 5],
       [9, 2, 3],
       [4, 5, 6]])

解决方案：

要实现所需的排序，请使用以下代码：

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

此代码按数组 a 第二列（列索引 1）中的值对数组 a 进行排序。关键在于 .argsort() 函数。它生成第二列中排序值的索引，然后用于相应地重新排列数组的行。

生成的数组将按第二列排序，行按升序排列基于该列中的值。

以上是如何按特定列对 NumPy 数组进行排序？的详细内容。更多信息请关注PHP中文网其他相关文章！