今天在学习tqdm
的时候,遇到一个问题,就是我发现trange
的输出跟主线程的print函数不是同步输出的;
我使用的测试代码如下:
import sys
from tqdm import trange
def main():
X = 100
for i in trange(X):
# for i in trange(X, file=sys.stdout):
print(i)
for j in range(X):
print(j)
k = j * i
print("Done️")
return 0
if __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️
之后才输出的,这让我们感觉很奇怪;
tqdm
在其底层实现中是使用子线程进行进度条的打印输出的也就是,进度条的打印输出是异步进行的,所以才会出现【100%|██████████| 100/100 [00:00<00:00,...]
进度条信息在Done️
】之后输出的情况;
tqdm
的进度条刷新也是通过end='\r'
的设置实现的我们如此猜测,是因为如果在中途进行print输出的话,就会重复显示进度条;我们使用了如下代码来进行测试:
import sys
from tqdm import trange
def main():
X = 10
k = 0
for i in trange(X, file=sys.stderr): # 使用默认的stderr输出
print(f"Outer loop {i}")
for j in range(2000000):
k += j * i
print(f"Done️: k = {k}")
return 0
if __name__ == '__main__':
main()
其输出信息如下:
0%| | 0/10 [00:00, ?it/s]Outer loop 0
Outer loop 1
20%|██ | 2/10 [00:00<00:00, 11.00it/s]Outer loop 2
Outer loop 3
40%|████ | 4/10 [00:00<00:00, 10.02it/s]Outer loop 4
Outer loop 5
60%|██████ | 6/10 [00:00<00:00, 9.67it/s]Outer loop 6
70%|███████ | 7/10 [00:00<00:00, 9.52it/s]Outer loop 7
Outer loop 8
90%|█████████ | 9/10 [00:00<00:00, 9.32it/s]Outer loop 9
100%|██████████| 10/10 [00:01<00:00, 9.53it/s]
Done️: k = 89999955000000
Process finished with exit code 0
可以看到中途会出现循环过程中的进度条信息;
【StackOverflow】Understanding Threading Behavior in tqdm for Progress Bar Updates