为何Python中PyMongo聚合结果的list()调用行为存在差异？

问题原因 #

PyMongo的aggregate()方法返回的不是普通列表，而是可迭代的游标对象，这个对象的核心特性是只能被遍历一次。遍历完成后，游标就会耗尽，再次调用list()或者尝试遍历都会得到空结果。

你提供的测试代码直接验证了这个特性：

out = list(output)  # 第一次遍历游标，将所有数据转成内存中的列表
print(out)  # 正常输出[{'_id': None, 'sum': 2700}]
print(list(output))  # 游标已耗尽，返回空列表[]

你最初的代码print(list(output)[0]["sum"])抛出IndexError，大概率是因为该游标之前已经被隐式遍历过（比如其他代码打印过output、调用过游标相关方法），导致转成列表时为空，自然取索引0会报错。

解决办法 #

核心思路：将游标一次性转换为列表并保存，后续所有操作都基于这个静态列表，不再重复操作原游标。

修改后的代码示例：

# 先把聚合游标转成列表并存储
output_list = list(get_mongo_collection().aggregate([
        {'$match': {'topic_name': topic, 'internal_id': self.internal_id }},
        {'$group': {'_id': None, 'sum': {'$sum': '$payload.val'}}}
]))

# 先判断列表非空，再取值，彻底避免IndexError
if output_list:
    print(output_list[0]["sum"])

补充说明 #

手动创建的普通列表是内存中的静态集合，可以重复遍历；而PyMongo游标是流式对象，从数据库分批拉取数据，遍历完成后不会再向数据库请求数据，这就是两者的本质差异。

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