NumPy数组排序:如何高效地按指定列排序?

碧海醫心
发布: 2025-03-02 18:30:53
原创
529人浏览过

numpy数组排序:如何高效地按指定列排序?

NumPy高效排序:按列排序详解

在数据分析中,对NumPy数组进行排序是常见操作。本文介绍如何利用NumPy库高效地对ndarray对象进行基于列的排序,特别是按指定列进行升序排序的方法。

假设有一个包含两列数据的NumPy数组,需要根据第二列的值进行升序排序。原始数组如下:

[[1577 2182]
 [1812 2181]
 [1813 2283]
 [1578 2285]]
登录后复制

排序后的预期结果:

[
[1812 2181]
[1577 2182]
[1813 2283]
[1578 2285]
]
登录后复制

NumPy的argsort()函数是实现此目标的关键。argsort()函数返回排序后的数组索引,而非排序后的数组本身。结合argsort()函数和数组索引,即可轻松实现按列排序。

以下代码演示了如何使用argsort()函数按第二列进行升序排序:

import numpy as np

arr = np.array([[1577, 2182],
                [1812, 2181],
                [1813, 2283],
                [1578, 2285]])

sorted_arr = arr[np.argsort(arr[:, 1])]
print(sorted_arr)
登录后复制

代码中,arr[:, 1]选择数组的第二列,np.argsort(arr[:, 1])返回第二列升序排序后的索引。然后,使用这些索引重新排列原始数组arr,得到按第二列升序排序的结果sorted_arr。 这种方法可以灵活地根据任何一列对多维数组进行排序。

以上就是NumPy数组排序:如何高效地按指定列排序?的详细内容,更多请关注php中文网其它相关文章!

最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习
PHP中文网抖音号
发现有趣的

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号