numpy.argpartition(a, kth, axis=-1, kind=‘introselect’, order=None)
(一)一般numpy中以arg开头的函数都是返回下标,而不改变原数组。
参数a是指传入的Numpy.array
参数kth是指列表中下标为k位置应该放置该数组中第k大的值
例1:
import numpy as np
arr = np.array([3, 9, 1, 0, 2, 1, 7, 5])
# 返回的是下标,由于python是从0开始计算而非1,所以kt