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

时间:2017-02-01 17:29:45

标签: python generator tqdm

我正在循环一个我知道长度的大文件,但由于它太大而无法放入内存中,所以我会懒得处理。我希望能够使用tqdm来跟踪我在文件中的进度,但由于它无法从我正在使用的生成器中获取示例总数,因此它显示的唯一内容是估计的迭代/第二。有没有办法告诉tqdm它将循环多少元素,所以我可以得到一些其他的统计数据?

1 个答案:

答案 0 :(得分:30)

您可以将长度传递给参数总计,以使其有效。

示例:

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