给 Python 添加进度条 | 给小白的 tqdm 精炼实例!
Python之王
共 3495字,需浏览 7分钟
· 2021-09-11
假设我们有一个循环:
for i in range(100):
do_something() # 这里做某些事
假设 do_something()
很慢,且运行时间不稳定,当我们运行时,我们并不知道运行到哪里了。
我们引入一个进度条。
from tqdm import trange
for i in trange(100):
do_something()
输出
5%|██▌ | 5/100 [00:05<01:35, 1.01s/it]
如上,我们就可用实时查看进度。
安装
这个库并不是 python 自带的库,使用 pip
安装就可。
pip install tqdm
来看看我的实验环境。
❯ python --version
Python 3.7.0
>>> tqdm.__version__
'4.59.0'
入门实例
说实在的,官方文档[1]写得并不是很好。
对于 range()
,我们把其改为 tqdm
中的 trange()
就可以。
其实, trange(5)
相当于 tqdm(range(5))
。
所以不难看出,用 tqdm 包裹 tqdm(可迭代的对象)
就可以自动为我们建立进度条。
>>> a = ['a', 'b', 'c'] * 10000
>>> from tqdm import tqdm
>>> for i in tqdm(a):
... pass
...
100%|███████████| 30000/30000 [00:00<?, ?it/s]
>>>
使用 for i in tqdm(a)
和 for i in a
逻辑上没有区别,但是前者让控制台多出了一个进度条。
自定义进度条
其实,我们还可用「手把手地」操作进度条。尤其是当我们的进度并不是单纯的依靠可迭代变量时。
假设我们有个任务如下。
import random
import time
class Task:
def __init__(self) -> None:
self.jobs = int(1e3)
@property
def job_done(self) -> bool:
return self.jobs <= 0
def do_job(self) -> int:
time.sleep(1)
job_minus = random.randint(1, 50)
self.jobs = max(0, self.jobs - job_minus)
return job_minus
我们总共有 1e3 = 1000
个任务要做,但是这些任务并不是每次做一个,而是每次可能做 [1, 50)
个中的任意数量的工作。
所以我们不可以单纯让 do_job
运行 1000 次,因为每次 do_job
的效率不同。
这时我们定制我们的进度条。
from tqdm import tqdm
task: Task = Task()
info = { 'efficiency': None }
with tqdm(
total=task.jobs, desc='Doing jobs'
) as t:
while not task.job_done:
job_minus = task.do_job()
info['efficiency'] = job_minus
t.update(job_minus)
t.set_postfix(info)
如上,我们声明了一个 tqdm
对象:
这个对象的计数总量是我们的总工作量 total=task.jobs
进度条的前缀是 'Doing jobs'
每一次刷新进度,我们进度条的增量是 t.update(job_minus)
我们进度条后缀输出信息 info['efficiency'] = job_minus
来看看效果。
❯ python .\example.py
Doing jobs: 9%|██▏ | 94/1000 [00:03<00:28, 31.73it/s, efficiency=30]
动图如下。
总结
通过实例探讨了 tqdm
进度条的使用。
参考资料
官方文档: https://tqdm.github.io/docs/tqdm/
评论
真高!比亚迪员工爆料比亚迪在越南的薪资水平:基本工资480万,全勤奖35万,交通补助20万,餐补110万,每周6天,每天10小时
上一篇:某大公司为逼迫员工离职,竟然把他的工位安排到厕所旁,没想到他直接开始记录领导的如厕时间,还发到公司大群...对此,你怎么看?--完--PS:欢迎在留言区留下你的观点,一起讨论提高。如果今天的文章让你有新的启发,欢迎转发分享给更多人。全文完,感谢你的耐心阅读。如果你还想看到我的文章,请一定给本
开发者全社区
0
某大公司为逼迫员工离职,竟然把他的工位安排到厕所旁,没想到他直接开始记录领导的如厕时间,还发到公司大群...
上一篇:字节的跳动职级与薪资(2024年)我们与公司间的合作,宛如两艘船只在茫茫大海上相互依靠,共同抵御风浪,携手驶向成功的彼岸。然而,当航向开始产生分歧,或是波涛汹涌的风浪改变了我们的初衷,我们或许应当冷静地选择和平分手,而非在风雨中硬撑。最近,一位网友的遭遇引起了广大职场人的关注和热议。这位网友
开发者全社区
0
金融研究 | 使用Python测量关键审计事项的「信息含量」
Tips: 公众号推送后内容只能更改一次,且只能改20字符。如果内容出问题,或者想更新内容, 只能重复推送。为了更好的阅读体验,建议阅读本文博客版, 链接地址https://textdata.cn/blog/2023-01-13-information-content-of-critical-aud
大邓和他的Python
0
我看阿里的年终奖总算发了!
到4月底了,这两天看朋友圈,发现阿里的年终奖终于发了,问了问老同学,也从网上检索了不少信息,基本搞清楚了阿里今年的年终奖情况。近来来阿里一些集团对绩效等级做了较大的调整,以前的旧绩效系统中,绩效分为3.25、3.5、3.75、4和5五个等级,其中4和5是较高绩效等级,较少见。而且之前3.5绩效内部划
公子龙
0
CVPR 2024|大视觉模型的开山之作!无需任何语言数据即可打造大视觉模型
↑ 点击蓝字 关注极市平台作者丨科技猛兽编辑丨极市平台极市导读 本文提出一种序列建模 (sequential modeling) 的方法,不使用任何语言数据,训练大视觉模型。>>加入极市CV技术交流群,走在计算机视觉的最前沿本文目录1 序列建模打造大视觉模型(来自 U
极市平台
1
金融研究(更新) | 使用Python构建关键审计事项的「信息含量」
Tips: 公众号推送后内容只能更改一次,且只能改20字符。如果内容出问题,或者想更新内容, 只能重复推送。为了更好的阅读体验,建议阅读本文博客版, 链接地址https://textdata.cn/blog/2023-01-13-information-content-of-critical-aud
大邓和他的Python
0
字节的跳动职级与薪资(2024年)
上一篇:阿里公布年终奖,P7, 3.5+,22W年终奖,还有35W长期现金激励,真香字节跳动自2012年3月成立以来,已经迅速成长为一个全球性的科技公司。其产品和服务已经遍布全球150多个国家与地区,并且支持超过75种不同的语言。在字节跳动的官方网站上,列出了一系列引人注目的产品和服务,包括但不限于
开发者全社区
0
盘点Lombok的几个骚操作,你绝对没用过!
👉 欢迎加入小哈的星球 ,你将获得: 专属的项目实战 / Java 学习路线 / 一对一提问 / 学习打卡 / 赠书福利全栈前后端分离博客项目 2.0 版本完结啦, 演示链接:http://116.62.199.48/ ,新项目正在酝酿中
小哈学Java
0