1 问题描述
今天在学习tqdm的时候,遇到一个问题,就是我发现trange的输出跟主线程的print函数不是同步输出的;
 我使用的测试代码如下:
import sysfrom tqdm import trangedef main():X = 100for i in trange(X):# for i in trange(X, file=sys.stdout):print(i)for j in range(X):print(j)k = j * iprint("Done🛠️")return 0if __name__ == '__main__':main()
控制台的输出如下:
# 输出信息很多,这里就只给出了最后几行信息
...
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
Done🛠️
100%|██████████| 100/100 [00:00<00:00, 3694.12it/s]Process finished with exit code 0
可以看到100%|██████████| 100/100 [00:00<00:00, 3694.12it/s]信息竟然是在Done🛠️之后才输出的,这让我们感觉很奇怪;
2 假设和猜测
(1)猜测:tqdm在其底层实现中是使用子线程进行进度条的打印输出的
 
也就是,进度条的打印输出是异步进行的,所以才会出现【100%|██████████| 100/100 [00:00<00:00,...]进度条信息在Done🛠️】之后输出的情况;
(2)猜测:tqdm的进度条刷新也是通过end='\r'的设置实现的
 
我们如此猜测,是因为如果在中途进行print输出的话,就会重复显示进度条;我们使用了如下代码来进行测试:
import sys
from tqdm import trangedef main():X = 10k = 0for i in trange(X, file=sys.stderr):  # 使用默认的stderr输出print(f"Outer loop {i}")for j in range(2000000):k += j * iprint(f"Done🛠️: k = {k}")return 0if __name__ == '__main__':main()
其输出信息如下:
  0%|          | 0/10 [00:00<?, ?it/s]Outer loop 0
Outer loop 120%|██        | 2/10 [00:00<00:00, 11.00it/s]Outer loop 2
Outer loop 340%|████      | 4/10 [00:00<00:00, 10.02it/s]Outer loop 4
Outer loop 560%|██████    | 6/10 [00:00<00:00,  9.67it/s]Outer loop 670%|███████   | 7/10 [00:00<00:00,  9.52it/s]Outer loop 7
Outer loop 890%|█████████ | 9/10 [00:00<00:00,  9.32it/s]Outer loop 9
100%|██████████| 10/10 [00:01<00:00,  9.53it/s]
Done🛠️: k = 89999955000000Process finished with exit code 0
可以看到中途会出现循环过程中的进度条信息;
3 提问备忘
【StackOverflow】Understanding Threading Behavior in tqdm for Progress Bar Updates