如何格式化 tqdm 进度条以显示每分钟而不是每秒的进度？

Question

我在GitHub或文档中找不到这个，但我想知道是否有一种集成的方式（首选）或解决方法来显示一次迭代所需的平均时间（以分钟而不是秒为单位）。

具体来说， tqdm 显示类似1283.31s/it但如果每次迭代需要几分钟或几小时，显示类似21m/it内容会更有帮助。

Answer 1

我们可以通过添加自定义为此bar_format参数的自定义实例tqdm使用说明 手册中得到部分描述：

class TqdmExtraFormat(tqdm):
    """Provides a `minutes per iteration` format parameter"""
    @property
    def format_dict(self):
        d = super(TqdmExtraFormat, self).format_dict
        rate_min = '{:.2f}'.format(1/d["rate"] / 60) if d["rate"] else '?'
        d.update(rate_min=(rate_min + ' min/' + d['unit']))
        return d

我们在这里所做的是使用自定义版本覆盖format_dict ，该版本添加了一个可以在bar_format参数中使用的新统计信息。 d是现有的format_dict ，因此我们可以使用它来获取{rate}的值，将其转换为每次迭代的时间（1/rate），除以 60，然后将其格式化为 2 个小数位。 然后，我们更新d与新的统计-我们甚至可以使用字符串连接添加一些格式。 请注意，在任何迭代完成之前， {rate}是None ，这意味着除非您执行if d["rate"]检查，否则这将产生错误。

完成后，如果您使用tqdm的自定义实例，我们可以在bar_format参数字符串中使用此统计信息bar_format参数中列出的参数：

b='{l_bar}{bar}| {n_fmt}/{total_fmt} [{elapsed}<{remaining}, {rate_min}{postfix}]'
for i in TqdmExtraFormat(range(2), unit_scale = 60, bar_format=b):
    time.sleep(6)