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
相关文章