随笔 - 384  文章 - 0  评论 - 35  阅读 - 142万

np.argsort()元素从小到大排序后,提取对应的索引index,可以一行搞定排序

函数用法

numpy.argsort(a, axis=-1, kind=’quicksort’, order=None)

功能: 将矩阵a按照axis排序,并返回排序后的下标,axis=0是列,1是行
参数: a:输入矩阵, axis:需要排序的维度
返回值: 输出排序后的下标

import numpy as np
x = np.array([1,4,3,-1,6,9])
x.argsort()
# array([3, 0, 1, 2, 4, 5], dtype=int64)

可以发现,argsort()是将X中的元素从小到大排序后,提取对应的原来的索引index,然后输出到y

复制代码
#所以取数组x的最小值的索引位置可以写成:
x[x.argsort()[0]]
#或者用argmin()函数
x[x.argmin()]

#数组x的最大值,写成:
x[x.argsort()[-1]]  # -1代表从后往前反向的索引
#或者用argmax()函数,不再详述
x[x.argmax()]
复制代码

输出排序后的数组

x[x.argsort()]
#或者
x[np.argsort(x)]

#array([-1,  1,  3,  4,  6,  9])

 那么给数据排序不就很简单了,直接一行就搞定了

posted on   小小喽啰  阅读(6931)  评论(0编辑  收藏  举报
编辑推荐:
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

点击右上角即可分享
微信分享提示