详细介绍Python进度条tqdm的使用

作者:修炼之路 时间:2022-12-13 19:45:08 

前言

有时候在使用Python处理比较耗时操作的时候,为了便于观察处理进度,这时候就需要通过进度条将处理情况进行可视化展示,以便我们能够及时了解情况。这对于第三方库非常丰富的Python来说,想要实现这一功能并不是什么难事。

tqdm就能非常完美的支持和解决这些问题,可以实时输出处理进度而且占用的CPU资源非常少,支持windowsLinuxmac等系统,支持循环处理多进程递归处理、还可以结合linux的命令来查看处理情况,等进度展示。

大家先看看tqdm的进度条效果

详细介绍Python进度条tqdm的使用

安装

github地址:https://github.com/tqdm/tqdm

想要安装tqdm也是非常简单的,通过pip或conda就可以安装,而且不需要安装其他的依赖库

pip安装


pip install tqdm

conda安装


conda install -c conda-forge tqdm

迭代对象处理

对于可以迭代的对象都可以使用下面这种方式,来实现可视化进度,非常方便


from tqdm import tqdm
import time

for i in tqdm(range(100)):
 time.sleep(0.1)
 pass

详细介绍Python进度条tqdm的使用

在使用tqdm的时候,可以将tqdm(range(100))替换为trange(100)代码如下


from tqdm import tqdm,trange
import time

for i in trange(100):
 time.sleep(0.1)
 pass

观察处理的数据

通过tqdm提供的set_description方法可以实时查看每次处理的数据


from tqdm import tqdm
import time

pbar = tqdm(["a","b","c","d"])
for c in pbar:
 time.sleep(1)
 pbar.set_description("Processing %s"%c)

详细介绍Python进度条tqdm的使用

手动设置处理的进度

通过update方法可以控制每次进度条更新的进度


from tqdm import tqdm
import time

#total参数设置进度条的总长度
with tqdm(total=100) as pbar:
 for i in range(100):
   time.sleep(0.05)
   #每次更新进度条的长度
   pbar.update(1)

详细介绍Python进度条tqdm的使用

除了使用with之外,还可以使用另外一种方法实现上面的效果


from tqdm import tqdm
import time

#total参数设置进度条的总长度
pbar = tqdm(total=100)
for i in range(100):
 time.sleep(0.05)
 #每次更新进度条的长度
 pbar.update(1)
#关闭占用的资源
pbar.close()

linux命令展示进度条

不使用tqdm


$ time find . -name '*.py' -type f -exec cat \{} \; | wc -l
857365

real  0m3.458s
user  0m0.274s
sys   0m3.325s

使用tqdm


$ time find . -name '*.py' -type f -exec cat \{} \; | tqdm | wc -l
857366it [00:03, 246471.31it/s]
857365

real  0m3.585s
user  0m0.862s
sys   0m3.358s

指定tqdm的参数控制进度条


$ find . -name '*.py' -type f -exec cat \{} \; |
 tqdm --unit loc --unit_scale --total 857366 >> /dev/null
100%|███████████████████████████████████| 857K/857K [00:04<00:00, 246Kloc/s]

$ 7z a -bd -r backup.7z docs/ | grep Compressing |
 tqdm --total $(find docs/ -type f | wc -l) --unit files >> backup.log
100%|███████████████████████████████▉| 8014/8014 [01:37<00:00, 82.29files/s]

自定义进度条显示信息

通过set_descriptionset_postfix方法设置进度条显示信息


from tqdm import trange
from random import random,randint
import time

with trange(100) as t:
 for i in t:
   #设置进度条左边显示的信息
   t.set_description("GEN %i"%i)
   #设置进度条右边显示的信息
   t.set_postfix(loss=random(),gen=randint(1,999),str="h",lst=[1,2])
   time.sleep(0.1)

详细介绍Python进度条tqdm的使用


from tqdm import tqdm
import time

with tqdm(total=10,bar_format="{postfix[0]}{postfix[1][value]:>9.3g}",
    postfix=["Batch",dict(value=0)]) as t:
 for i in range(10):
   time.sleep(0.05)
   t.postfix[1]["value"] = i / 2
   t.update()

详细介绍Python进度条tqdm的使用

多层循环进度条

通过tqdm也可以很简单的实现嵌套循环进度条的展示


from tqdm import tqdm
import time

for i in tqdm(range(20), ascii=True,desc="1st loop"):
 for j in tqdm(range(10), ascii=True,desc="2nd loop"):
   time.sleep(0.01)

详细介绍Python进度条tqdm的使用

pycharm中执行以上代码的时候,会出现进度条位置错乱,目前官方并没有给出好的解决方案,这是由于pycharm不支持某些字符导致的,不过可以将上面的代码保存为脚本然后在命令行中执行,效果如下

详细介绍Python进度条tqdm的使用

多进程进度条

在使用多进程处理任务的时候,通过tqdm可以实时查看每一个进程任务的处理情况


from time import sleep
from tqdm import trange, tqdm
from multiprocessing import Pool, freeze_support, RLock

L = list(range(9))

def progresser(n):
 interval = 0.001 / (n + 2)
 total = 5000
 text = "#{}, est. {:<04.2}s".format(n, interval * total)
 for i in trange(total, desc=text, position=n,ascii=True):
   sleep(interval)

if __name__ == '__main__':
 freeze_support() # for Windows support
 p = Pool(len(L),
      # again, for Windows support
      initializer=tqdm.set_lock, initargs=(RLock(),))
 p.map(progresser, L)
 print("\n" * (len(L) - 2))

详细介绍Python进度条tqdm的使用

pandas中使用tqdm


import pandas as pd
import numpy as np
from tqdm import tqdm

df = pd.DataFrame(np.random.randint(0, 100, (100000, 6)))

tqdm.pandas(desc="my bar!")
df.progress_apply(lambda x: x**2)

详细介绍Python进度条tqdm的使用

递归使用进度条


from tqdm import tqdm
import os.path

def find_files_recursively(path, show_progress=True):
 files = []
 # total=1 assumes `path` is a file
 t = tqdm(total=1, unit="file", disable=not show_progress)
 if not os.path.exists(path):
   raise IOError("Cannot find:" + path)

def append_found_file(f):
   files.append(f)
   t.update()

def list_found_dir(path):
   """returns os.listdir(path) assuming os.path.isdir(path)"""
   try:
     listing = os.listdir(path)
   except:
     return []
   # subtract 1 since a "file" we found was actually this directory
   t.total += len(listing) - 1
   # fancy way to give info without forcing a refresh
   t.set_postfix(dir=path[-10:], refresh=False)
   t.update(0) # may trigger a refresh
   return listing

def recursively_search(path):
   if os.path.isdir(path):
     for f in list_found_dir(path):
       recursively_search(os.path.join(path, f))
   else:
     append_found_file(path)

recursively_search(path)
 t.set_postfix(dir=path)
 t.close()
 return files

find_files_recursively("E:/")

详细介绍Python进度条tqdm的使用

注意

在使用tqdm显示进度条的时候,如果代码中存在print可能会导致输出多行进度条,此时可以将print语句改为tqdm.write,代码如下


for i in tqdm(range(10),ascii=True):
 tqdm.write("come on")
 time.sleep(0.1)

来源:https://blog.csdn.net/sinat_29957455/article/details/97558787

标签:Python,进度条,tqdm
0
投稿

猜你喜欢

  • 一文掌握MySQL表的创建和约束

    2024-01-25 18:36:53
  • php 多个变量指向同一个引用($b = &$a)用法分析

    2023-11-02 23:28:11
  • ASP Google的translate API代码

    2011-04-03 11:16:00
  • sqlserver 多表查询不同数据库服务器上的表

    2012-04-13 11:41:51
  • Go实现MD5加密的三种方法小结

    2024-02-08 03:47:37
  • 在Linux系统上部署Apache+Python+Django+MySQL环境

    2024-01-14 00:51:15
  • python中open函数的基本用法示例

    2021-08-03 20:42:31
  • Python 线程池模块之多线程操作代码

    2022-06-14 06:01:47
  • python爬虫库scrapy简单使用实例详解

    2022-03-22 23:46:11
  • DNS优化的原理和方法

    2012-04-26 16:43:56
  • 如何使用Python+ChatGPT批量生成论文

    2023-10-01 20:30:17
  • T-SQL问题解决集锦 数据加解密全集

    2012-07-11 15:34:08
  • Tensorflow 查看变量的值方法

    2022-12-14 09:36:47
  • 在javascript中,null>=0 为真,null==0却为假,null的值详解

    2024-04-25 13:08:52
  • TF-IDF算法解析与Python实现方法详解

    2021-12-28 02:38:47
  • python3安装speech语音模块的方法

    2023-03-24 12:09:32
  • python中列表添加元素的几种方式(+、append()、extend())

    2022-07-17 19:59:42
  • MySQL回表的性能伤害程度有多大

    2024-01-29 10:55:55
  • 微信小程序获取当前位置的详细步骤

    2024-04-08 10:52:09
  • python3中的logging记录日志实现过程及封装成类的操作

    2023-07-30 21:58:21
  • asp之家 网络编程 m.aspxhome.com