python - tqdm 在较大循环内具有间隔

Question

我有一个外循环

for i in itertools.count():

并且有间隔发生的事件：

if i % log_interval == 0:
    perform_logging()
if i % save_interval == 0:
    save()
# etc.

我想使用 tqdm 来跟踪下一个日志/保存/等的进度。理想情况下，tqdm 会同时打印多个进度条，例如

最好的方法是什么？谢谢！

标签： pythontqdm

Answer 1

像这样的东西：

from __future__ import division, print_function
import itertools
from math import ceil
from tqdm import tqdm
from time import sleep

total = 987
log_interval = 2
save_interval = 9
with tqdm(total=total, desc="overall") as tOverall:
  with tqdm(total=ceil(total / log_interval), unit="log") as tLog:
    with tqdm(total=ceil(total / save_interval), unit="save") as tSave:
      for i in itertools.count(0, 1):
        sleep(0.01)
        if i % log_interval == 0:
          perform_logging()
          tLog.update()
        if i % save_interval == 0:
          save()
          tSave.update()
        if i + 1 == total:
          break
        tOverall.update()
print('\n')