当前位置:   article > 正文

Numpy中的argsort源码解析_python argsort源代码

python argsort源代码

numpy的argsort排序算法,默认为快速排序模式,但其中是由三个算法组成的,分别为快速排序,插入排序以及堆排序,官方给的阈值为15,即需要排序的元素长度超过15则会进入快速排序。

快速排序的细节:前后循环到中间位置pm交换值使得pm前面的值小于pm对应值,后面的值大于pm对应的值,然后分为两部分,把元素较多部分放入栈中,利用栈来进行深度遍历。

之后,再进入插入排序,双指针往后退来排序。

最后,如果达成某个条件(NPY_UNLIKELY(cdepth < 0))进入堆排序,这个还没看懂,至少我还进入过这个排序- -

参考资料:
https://github.com/numpy/numpy/blob/v1.20.0/numpy/core/src/multiarray/methods.c
https://github.com/numpy/numpy/blob/v1.20.0/numpy/core/src/npysort/quicksort.c.src
https://wiki.jikexueyuan.com/project/easy-learn-algorithm/fast-sort.html

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/638667
推荐阅读
相关标签
  

闽ICP备14008679号