tqdm显示我知道长度的生成器的进度

2022-03-16 00:00:00 python generator tqdm

问题描述

我正在循环一个大文件,我知道它的长度,但由于它太大而无法放入内存,所以处理速度很慢。我希望能够使用tqdm跟踪整个文件的进度,但是由于它不能从我正在使用的生成器中获得示例的总数,所以它唯一显示的就是估计的迭代次数/秒。有没有办法告诉tqdm它将在Total上循环多少个元素,以便我可以获得其他一些统计信息?


解决方案

您可以将总长度传递给参数总数以使其正常工作。

示例:

from tqdm import tqdm

length = 1000000
generator = (3 * n for n in range(length))  # just doing something random
for n in tqdm(generator, total=length):
    pass

相关文章