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

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

DDD
发布: 2024-12-25 08:56:09
原创
711 人浏览过

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

按列对 NumPy 数组进行排序

按特定列对 NumPy 数组进行排序通常是数据操作的必要操作。例如,考虑数组:

a = array([[9, 2, 3],
           [4, 5, 6],
           [7, 0, 5]])
登录后复制

假设我们要根据第二列中的值对 a 的行重新排序。为了实现这一点,我们可以利用 NumPy 中的 argsort() 函数。

a[:, 1].argsort()
登录后复制

上面的表达式将 argsort() 应用于 a 的第二列,生成与排序值相对应的索引数组。具体来说,它返回:

array([1, 0, 2])
登录后复制

这些索引表示 a 的行应重新排序以按第二列升序排序的顺序。结合数组索引,我们可以得到排序后的数组:

a[a[:, 1].argsort()]
登录后复制

该表达式返回:

array([[7, 0, 5],
       [9, 2, 3],
       [4, 5, 6]])
登录后复制

根据需要,a 的行已根据第二列排序.

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

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