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

Question

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

Answer 1

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