当然可行！tqdm自动识别迭代器总数的核心逻辑是：它会主动尝试读取迭代器的长度相关信息，不需要你手动传total=参数，只要给自定义迭代器添加对应的方法就行，完全符合你要的优雅感。

具体分两种场景来实现：

1. 提前知道确切总数：实现__len__()方法 #

如果你的迭代器能提前确定最终会产出多少个元素，直接给迭代器类加上__len__()方法就行。tqdm会优先读取这个方法的返回值作为总进度数。

举个简单的例子：

from tqdm import tqdm

class MyCustomIterator:
    def __init__(self, dataset):
        self.dataset = dataset
        self.position = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.position >= len(self.dataset):
            raise StopIteration
        item = self.dataset[self.position]
        self.position += 1
        return item

    # 关键：添加__len__方法返回总数
    def __len__(self):
        return len(self.dataset)

# 使用时tqdm会自动识别总数
for item in tqdm(MyCustomIterator([1, 2, 3, 4, 5])):
    # 你的业务逻辑
    pass

2. 无法确定确切总数，但能给出估计值：实现__length_hint__()方法 #

如果你的迭代器是懒加载的，没法提前知道精确总数，但能给出一个合理的元素数量估计（比如剩余待处理的元素数），可以实现PEP 424定义的__length_hint__()方法。tqdm会读取这个提示值来显示进度条（虽然是估计值，但用户体验比没有进度条好太多）。

示例代码：

from tqdm import tqdm

class MyLazyIterator:
    def __init__(self, max_items):
        self.max_items = max_items
        self.current = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.current >= self.max_items:
            raise StopIteration
        self.current += 1
        return self.current

    # 关键：添加长度提示方法
    def __length_hint__(self):
        # 返回剩余元素的估计数量
        return self.max_items - self.current

# tqdm会自动读取长度提示来显示进度
for num in tqdm(MyLazyIterator(100)):
    # 你的业务逻辑
    pass

补充说明 #

tqdm读取长度信息的优先级是：__len__() > __length_hint__() > 手动传total=。所以只要你的自定义迭代器实现了其中一个方法，tqdm就会自动识别总数，不需要额外传参啦。

内容的提问来源于stack exchange，提问作者Duane