0
点赞
收藏
分享

微信扫一扫

np.argsort()用法

_阿瑶 2022-02-05 阅读 84
  • np.argsort()用法

np.argsort(a, axis=-1, kind=’quicksort’, order=None)
功能: 将矩阵a按照axis排序,并返回排序后的下标
参数: a:输入矩阵, axis:需要排序的维度
返回值: 输出排序后的下标

举例:(一维数组)

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

argsort()是将X中的元素从小到大排序后,提取对应的索引index,然后输出到y
如x[3]=-1最小,x[5]=9最大

所以取数组x的最小值可以写成:x[x.argsort()[0]]

举报

相关推荐

0 条评论