Python---返回列表Top K值索引

一个叫欧维的程序员在此写博客

于 2021-08-16 15:08:11 发布

阅读量1.1k

点赞数 1

分类专栏： Python 文章标签： python numpy

本文链接：https://blog.csdn.net/qq_41982466/article/details/119733613

版权

Python 专栏收录该内容

56 篇文章

订阅专栏

该篇博客介绍了如何使用Python的numpy库来找到列表中最大的K个元素的索引。示例代码展示了如何定义一个名为top_K_idx的函数，它接受一个数据列表和整数K作为参数，返回列表中最大K个元素的索引。通过调用该函数并传入示例列表s和K值3，返回了列表s中最大三个元素的索引。

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

Python—返回列表Top K值索引

import numpy as np

def top_K_idx(data, k):
    data = np.array(data)
    idx = data.argsort()[-k:][::-1]
    return idx

k = 3
s = [1,3,4,2,5]
top_K_idx(s, k)

返回结果：
在这里插入图片描述