首页 > 后端开发 > Python教程 > 如何按特定列对 NumPy 数组进行排序?

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

DDD
发布: 2024-11-29 18:49:13
原创
792 人浏览过

How Do I Sort a NumPy Array by a Specific Column?

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

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板