按列对 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中文网其他相关文章!