numpy.argsort详解

numpy.argsort

  • 用例:
    numpy.argsort(a, axis=-1, kind=None, order=None)

  • 功能:
    返回数组排序后的元素索引值。
    根据kind指定的算法对数组沿着axis轴进行排序。其返回值的形状和a一致,返回值内容为排序后元素在原始数组中的索引。

  • 参数

变量名 数据类型 功能
a 数组型变量 被排序的数组。
axis 整型或None,可选参数 指定沿着哪个轴排序。默认值是-1即最后一维。若为None,则会将数组拉伸为一维。
kind {‘quicksort’, ‘mergesort’, ‘heapsort’, ‘stable’}, 可选参数 排序算法,默认为’quicksort’。注意:'stable’和’mergesort’的后端都是使用timsort算法,并且在通常情况下,
算法的计算结果会随数值类型的不同而发生改变。保留’mergesort’参数是为了向后兼容。
order 字符串、字符串组成的列表,可选参数 当数组a定义了字段时,此参数指定先比较哪个字段再比较哪个字段。通过字符串可以单一指定某个字段,
并且并不是所有字段均需被指定,未被指定的字段也会参与排序(按照他们在dtype中的出现顺序进行排序)。
  • 返回值
变量名 数据类型 功能
index_array n维数组或整数 将a沿着指定轴排序后其索引组成的数组。若a为一维数组,a[index_array]生成排序后的a数组。
更一般的,无论a为多少维数组,我们均可使用np.take_along_axis(a, index_array, axis=axis)生成排序后的a数组
  • 备注
    有关不同排序算法的说明,请参阅sort
    NumPy 1.4.0及以后的版本中,argsort可以对包含实数、复数、空值的数组进行排序。在sort中详细阐述了增强的排序算法。

  • 示例:

import numpy as np
  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值