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