获取np数组前N大值索引
较新的 NumPy 版本(1.8 及更高版本)为此有一个名为argpartition的函数。 要获得四个最大元素的索引,请执行
>>> a = np.array([9, 4, 4, 3, 3, 9, 0, 4, 6, 0])
>>> a
array([9, 4, 4, 3, 3, 9, 0, 4, 6, 0])
>>> ind = np.argpartition(a, -4)[-4:]
>>> ind
array([1, 5, 8,
原创
2021-10-08 16:22:50 ·
1104 阅读 ·
0 评论