Numpy的argpartion函数

lscelory

于 2019-11-17 00:32:17 发布

阅读量2.4k

点赞数

分类专栏： Python 文章标签： Numpy

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/lscelory/article/details/103104526

版权

Numpy的argpartition函数基于快排，返回数组中元素划分后的索引，可用于高效获取最大的N个元素，如top5。它不对原数组排序，仅返回索引，再通过索引取出最大值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Numpy的argpartion函数

一、np.argpartion()

Numpy的argpartion函数来源于快排算法中的一个典型操作partition，即根据一个数值x，把数组中的元素划分成两半，使得index前面的元素都不大于x，index后面的元素都不小于x。
np.argpartition不对原数组操作，只返回一个按照上述操作排序过后的index。通过这个函数可以高效地找到 N 个最大值的索引并返回 N 个值。在给出索引后，我们可以根据需要进行值排序。

import numpy as np
array = np.array([10, 7, 4, 3,

最低0.47元/天解锁文章

评论

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。