MacwinWin的博客

原创 numpy中的argpartition用法

毕业论文写完了，闲来无事搞了一个汉字书法识别比赛，提交的结果需要是Top-5，在google排序方法的时候，发现一个比argsort效率更高的方法：argpartition$ import numpy as np$ a = np.array([9, 4, 4, 3, 3, 9, 0, 4, 6, 0])$ print(np.argpartition(a, 4)) #将数组a中所有元素（包括重复...