python 如何在list中找Topk的数值和索引
需求：
对于一个python list 或者numpy数组，我需要找到这个list中最大的K个数及其对应的下标。
解决方式：
1. 可以构造字典通过排序解决，不过代码量较多。
2. 使用heapq库，可以直接获取最大值的下标和数值。
import heapq a = [4,2,6,1,9,9] # 获取下标， 输出为[4, 5, 2] heapq.nlargest(3, range(len(a)), a.__getitem__) # 获取数值， 输出为[9, 9, 6] heapq.nlargest(3,a)
如果要取最小的数，使用 nsmallest即可
补充：Python 利用中间值求TopK 算法
算法思想
首先我们要思考，我要做什么？解决什么问题？
TopK问题，找出一组数据中的前K个最大值或者最小值，这个数据是否重复？要做去重处理？
ok 我们明确我们做什么了 ，那介绍的python处理的topK 算法过程是怎么样的呢？
如果用排序那就没必要引入topK 了，当数据强大的时候选取TopK 可以省略很多排序的计算，至于有多优化自己去思考下，就比如排列组合的C，A的区别，一个是抽取，一个是抽取并排列…
以下以找出TopK 的最大值为例，最小值的可以自己修改一下下就可以
介绍的算法思想是利用中间值，将数列分为三部分 ，
【比中间值大的列表】，中间值，【比中间值小的列表】
那么我们当比较
【比中间值大的列表】的个数 == k
的时候就可以得出前K个最大值了，因此
重点就是找出这个中间值
如何找出中间值
以列表的第一个数开始为中间值，拆分为三部分
if 【比中间值大的列表】的个数 == k：return 中间值 #程序出口，结束。
if 【比中间值大的列表】的个数 < k ：
·····继续在【比中间值小的列表】找
·····K - 【比中间值大的列表】的个数 -1 个数
（为什么要减一，1是前一次的中间值，分的三部分，前部分后部分都没有包含中间值，因此…）
if 【比中间值大的列表】的个数 > k ：
…也就是说比中间值大的列表比K还大，那就在这个列表中继续找就行
结合代码和注释看
如果要找最小值，只需要改一下就ok ,还可以设置一个布尔值的输入，来做前K个最大值最小值
#2019 11 04
#author 半斤地瓜烧
#TopK 算法，找出序列中前K个最大值的
#输入一个seq
# 输出以seq[0]为中间值 划分的三个部分，中间值，比这个值大的seq ,比这个值小的seq，
# 即splitNum,theBig,theSmall
def Split_Seq(seq):
splitNum = seq[0]
seq = seq[1:]#两个部分都不包含中间值，因此切片去除seq[0]
theBig = [x for x in seq if x >= splitNum]
theSmall = [x for x in seq if x < splitNum]
return splitNum,theBig,theSmall
#找出中间值
def topKNum(seq,k):
splitNum, theBig, theSmall = Split_Seq(seq)
theBigLen = len(theBig)
if k == theBigLen:
return splitNum#出口,返回这个中间值,
# 为什么不直接返回thebig?因为存在递归的原因thebig 不是在初始的seq找出来的
#需要重新Split,即可，读者自己思考
# 大值的列表中还未够K个数的情况，
if k > theBigLen:
return topKNum(theSmall,k-theBigLen-1)
# 大值的列表中大于K个数的情况
return topKNum(theBig,k)
#由中间值找出TopK个值，<list>
def getTopK(seq,k):
return [i for i in seq if i > topKNum(seq, k)]
if __name__ == '__main__':
alist = [7, 3, 5, 1,885,234,2211,222,22, 2, 11, 2, 115]
print("===为了验证，引入排序观看===", sorted(alist,reverse= True))
print(getTopK(alist, 3))
以上为个人经验，希望能给大家一个参考，也希望大家多多支持我们。
相关推荐
-
python 的topk算法实例
我就废话不多说了,还是直接看代码吧! #! conding:utf-8 def quick_index(array, start, end): left, right = start, end key = array[left] while left < right: while left < right and array[right] > key: right -= 1 array[left] = array[right] while left < right and arra
-
python list 查询是否存在并且并返回下标的操作
看代码吧~ def find_all_index(arr, item): return [i for i, a in enumerate(arr) if a == item] if __name__ == '__main__': print(find_all_index([1,2,3,4,4,3,89],4)) 输出结果: 3,4 补充:python字符串\列表中查找出某个值且对应的下标 1.字符串中查找值和对应的下标 a = "123yui78y8y67tuy" print re.f
-
python topk()函数求最大和最小值实例
函数介绍 a.topk()求a中的最大值或最小值,返回两个值,一个是a中的值(最大或最小),一个是这个值的索引. 代码示例 >>> import torch >>> a=torch.randn((3,5)) >>> a tensor([[-0.4790, -0.6308, 0.2370, 0.0380, -0.0579], [-0.6712, -3.5483, -0.2370, -0.8658, 0.4145], [-1.4126, -0.8786,
-
解决python列表list中的截取问题
List(列表)作为python中使用最频繁的数据类型,如果能够把列表掌握,那么对于Python的掌握是有很大帮助的. 并且列表的元素的值是可以修改的 List的格式:(列表中的元素可以是字符串类型,也可以是数字类型,布尔型等等) #Author:LJZ list=['123','abc',0,True] for i in range(4): x=list[i] print(x) 执行结果: 123 abc 0 True 对于列表的截取操作(这个操作里面有一些细节,下面我总结了一下) 注意:列表
-
Python中移除List重复项的五种方法
本文列些处几种去除在Python 列表中(list)可能存在的重复项,这在很多应用程序中都会遇到的需求,作为程序员最好了解其中的几种方法 以备在用到时能够写出有效的程序. 方法1:朴素方法 这种方式是在遍历整个list的基础上,将第一个出现的元素添加在新的列表中. 示例代码: # Python 3 code to demonstrate # removing duplicated from list # using naive methods # initializing list test_l
-
python 如何在list中找Topk的数值和索引
需求: 对于一个python list 或者numpy数组,我需要找到这个list中最大的K个数及其对应的下标. 解决方式: 1. 可以构造字典通过排序解决,不过代码量较多. 2. 使用heapq库,可以直接获取最大值的下标和数值. import heapq a = [4,2,6,1,9,9] # 获取下标, 输出为[4, 5, 2] heapq.nlargest(3, range(len(a)), a.__getitem__) # 获取数值, 输出为[9, 9, 6] heapq.nlarge
-
Python如何在main中调用函数内的函数方式
一般在Python中在函数中定义的函数是不能直接调用的,但是如果要用的话怎么办呢? 一般情况下: def a():#第一层函数 def b():#第二层函数 print('打开文件B') b()#第二层中的函数直接调用 结果显示: Traceback (most recent call last): File "C:/Users/rog/Desktop/wenzhang.py", line 4, in <module> b() NameError: name 'b' is
-
python 如何在 Matplotlib 中绘制垂直线
介绍 Matplotlib是Python中使用最广泛的数据可视化库之一.Matplotlib的受欢迎程度大部分来自其自定义选项.您可以调整其对象层次结构中的几乎任何元素. 在本教程中,我们将研究如何在Matplotlib图上绘制垂直线,这使我们能够标记和突出显示图的某些区域,而无需缩放或更改轴范围. 创建图 让我们首先用一些随机数据创建一个简单的图: import matplotlib.pyplot as plt import numpy as np fig, ax = plt.subplots
-
python如何在word中存储本地图片
想要利用Python来操作word文档可以使用docx模块. 安装: pip install python-docx from docx import Document from docx.shared import Inches string = '文字内容' images = '1.jpg' # 保存在本地的图片 doc = Document() # doc对象 doc.add_paragraph(string) # 添加文字 doc.add_picture(images, width=In
-
使用Python对mongo数据库中字符串型正负数值比较大小
数据库中数据展示: 使用python代码实现: # Requires pymongo 3.6.0+ from pymongo import MongoClient client = MongoClient("mongodb://root:88888888@192.168.124.49:27017") database = client["test-mongo"] collection = database["students2"] # Creat
-
python实现从字符串中找出字符1的位置以及个数的方法
本文实例主要实现给出任意字符串,获取字符串中某字符的位置以及出现的总次数. 实现该功能代码的时候可以使用函数enumerate来将字符串分离成位置和字符,然后进行比较即可. 具体实现代码如下: #!/bin/env python #-*- coding:utf-8 -*- # """ 用enumerate将string中的1都找出来, 用enumerate实现: """ def get_1_pos(string): onePos=[] try:
-
详解python如何在django中为用户模型添加自定义权限
django自带的认证系统能够很好的实现如登录.登出.创建用户.创建超级用户.修改密码等复杂操作,并且实现了用户组.组权限.用户权限等复杂结构,使用自带的认证系统就能帮助我们实现自定义的权限系统达到权限控制的目的. 在django中默认情况下,syncdb运行时安装了django.contrib.auth,它会为每个模型创建默认权限,如foo.can_change,foo.can_delete和foo.can_add.要向模型添加自定义权限,可以添加类Meta:在模型下,并在其中定义权限,如此处
-
Python脚本如何在bilibili中查找弹幕发送者
总所周知bilibili是没有办法直接查看弹幕的发送者的,这使得当我们看到一些nt弹幕的时候虽然生气,却无可奈何,但是B站是可以屏蔽某个用户发送的弹幕的,这说明数据接口里肯定有用户信息,由于最近在学爬虫,所以我想先找找弹幕接口,分析下里面的数据. 找接口 找接口当然是随便打开一个视频然后F12啦,可是当我找了两圈后我傻眼了,没找到啊..得,不能把时间浪费在这种事情上,果断打开百度,不出所料,找到了如下的两个接口,都是XML格式网页 https://comment.bilibili.com/+ci
-
如何在C++中调用Python
Python的安装 为了使用Python.h这个扩展项,我们需要安装一个python*-dev而不是python*,这两者略有区别,下面的案例展示的是在Ubuntu20.04下安装python3.9-dev的方法: dechin@ubuntu2004:~/projects/gitlab/dechin/$ sudo apt install python3.9-dev 正在读取软件包列表... 完成 正在分析软件包的依赖关系树 正在读取状态信息... 完成 下列软件包是自动安装的并且现在不需要了:
-
python实现从一组颜色中找出与给定颜色最接近颜色的方法
本文实例讲述了python实现从一组颜色中找出与给定颜色最接近颜色的方法.分享给大家供大家参考.具体分析如下: 这段代码非常有用,可以找到指定颜色相似的颜色,比如有一组8个颜色,现在给定一个rgb格式的演示,找出它与8个颜色中的哪一个最接近,如果你需要做一个按照图片颜色搜索图片的程序,这个就非常有用了. 复制代码 代码如下: from colorsys import rgb_to_hsv colors = dict(( ((196, 2, 51), "RED"), ((255, 165