python中update,numpy中的argpartition

 2023-10-21 阅读 18 评论 0

摘要:numpy.argpartition(a, kth, axis=-1, kind='introselect', order=None) 在快排算法中,有一個典型的操作:partition。這個操作指:根據一個數值x,把數組中的元素劃分成兩半,使得index前面的元素都不大于x,inde

numpy.argpartition(a, kth, axis=-1, kind='introselect', order=None)

在快排算法中,有一個典型的操作:partition。這個操作指:根據一個數值x,把數組中的元素劃分成兩半,使得index前面的元素都不大于x,index后面的元素都不小于x。

python中update。numpy中的argpartition()函數就是起的這個作用。對于傳入的數組a,先用O(n)復雜度求出第k大的數字,然后利用這個第k大的數字將數組a劃分成兩半。

此函數不對原數組進行操作,它只返回分區之后的下標。一般numpy中以arg開頭的函數都是返回下標而不改變原數組。

此函數還有另外兩個參數:

  • kind:用于指定partition的算法
  • order:表示排序的key,也就是按哪些字段進行排序

python中的del函數。當我們只關心topK時,我們不需要使用np.sort()對數組進行全量排序,np.argpartition()已經夠用了。

參考資料

https://docs.scipy.org/doc/numpy/reference/generated/numpy.argpartition.html

轉載于:https://www.cnblogs.com/weiyinfu/p/9794551.html

版权声明:本站所有资料均为网友推荐收集整理而来,仅供学习和研究交流使用。

原文链接:https://hbdhgg.com/1/153599.html

发表评论:

本站为非赢利网站,部分文章来源或改编自互联网及其他公众平台,主要目的在于分享信息,版权归原作者所有,内容仅供读者参考,如有侵权请联系我们删除!

Copyright © 2022 匯編語言學習筆記 Inc. 保留所有权利。

底部版权信息