numpy.argpartition(a, kth, axis=-1, kind='introselect', order=None)
在快排算法中，有一個典型的操作：partition。這個操作指：根據一個數值x，把數組中的元素划分成兩半，使得index前面的元素都不大於x，index后面的元素都不小於x。
numpy中的argpartition()函數就是起的這個作用。對於傳入的數組a，先用O(n)復雜度求出第k大的數字，然后利用這個第k大的數字將數組a划分成兩半。
此函數不對原數組進行操作，它只返回分區之后的下標。一般numpy中以arg開頭的函數都是返回下標而不改變原數組。
此函數還有另外兩個參數：
- kind：用於指定partition的算法
- order：表示排序的key，也就是按哪些字段進行排序
當我們只關心topK時，我們不需要使用np.sort()對數組進行全量排序，np.argpartition()已經夠用了。
參考資料
https://docs.scipy.org/doc/numpy/reference/generated/numpy.argpartition.html