Python np.argsort() 函数的用法

np.argsort 函数的用法

调用方法

python 复制代码
numpy.argsort(a, axis=-1, kind=None, order=None)

各个参数意义:
a:输入数组,该数组将会被排序。
axis:沿着哪个轴排序,默认轴为 -1,即最后一个轴。如果为 None,那么会使用 flatten 后的数组。
axis:可选参数,选择需要做差的轴,默认的是最后的轴。
kind:排序方法,可以选择 {'quicksort', 'mergesort', 'heapsort', 'stable'},这里不做过多介绍。
order:当 a 是一个数组时,这个值制定了

返回值:返回一个数组 a 排序的索引值。

示例1

python 复制代码
import numpy as np

x = np.array([3, 1, 2])
print(np.argsort(x))
"""
result:
[1 2 0]
"""

可以看到输出的结果对应排序后的索引值,原数组 x 经过排序后的顺序从小到大应该为 [1, 2, 3],因为原始 x 数组中元素为 [3, 1, 2],即元素 1 的索引值为 1, 元素 2 的索引值为 2, 元素 3 的索引值为 0。因此,对应的原始 x 数组中元素的索引值顺序为 [1, 2, 0]

示例2

python 复制代码
import numpy as np

x = np.array([[0, 3], [2, 2]])
print(np.argsort(x, axis=0))
"""
result:
[[0 1]
 [1 0]]
"""

axis = 0 表示沿着 y 轴排序,因此得到上述结果。不理解的小伙伴可以阅读numpy数组的坐标轴问题

如果大家觉得有用,就点个赞让更多的人看到吧~

相关推荐
紫小米19 小时前
后端日志管理
python·fastapi
你不是我我19 小时前
【Java 开发日记】HTTP3 性能更好,为什么内网微服务依然多用 HTTP2?HTTP2 内网优势是什么?
java·开发语言·微服务
tjl521314_2120 小时前
04C++ 名称空间(Namespace)
开发语言·c++
赏金术士20 小时前
Kotlin 数据流与单双向绑定
android·开发语言·kotlin
白雪茫茫20 小时前
监督学习、半监督学习、无监督学习算法详解
python·学习·算法·ai
つ安静与叛逆的小籹人20 小时前
小红书API:通过笔记ID获取笔记详情数据教程
笔记·python
05候补工程师20 小时前
[实战复盘] 拒绝 AI 屎山!我从设计模式中学到的“调教”AI 新范式
人工智能·python·设计模式·ai·ai编程
逻辑驱动的ken21 小时前
Java高频面试场景题25
java·开发语言·深度学习·面试·职场和发展
AI人工智能+电脑小能手1 天前
【大白话说Java面试题】【Java基础篇】第32题:Java的异常处理机制是什么
java·开发语言·后端·面试
阿豪只会阿巴1 天前
【没事学点啥】TurboBlog轻量级个人博客项目——项目介绍
javascript·python·django·html