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

Question

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

Answer 1

您可以将长度传递给参数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