Flask+pymongo场景下如何将MongoDB聚合结果转为单对象键值对？

1. 调整MongoDB聚合查询逻辑 #

你需要在聚合管道里完成数组拆分、统计、结果合并三步操作，直接在数据库层面把零散的统计结果整合成单个对象，避免后续额外做格式转换：

• 用$unwind拆分categories数组，把每个分类项拆为独立文档
• 用$group按分类名称分组，统计每个分类的出现次数
• 二次$group把所有统计项整理为键值对数组，再通过$arrayToObject转成单个对象

2. 完整代码实现 #

from flask import jsonify
import pymongo

# 假设你已经完成pymongo初始化，餐厅集合对象为restaurants_col
pipeline = [
    # 拆分分类数组
    {"$unwind": "$categories"},
    # 按分类分组统计数量
    {"$group": {
        "_id": "$categories",
        "count": {"$sum": 1}
    }},
    # 把所有统计结果整理为k-v格式的数组
    {"$group": {
        "_id": None,
        "count_list": {"$push": {"k": "$_id", "v": "$count"}}
    }},
    # 把k-v数组转换为单个分类统计对象
    {"$replaceRoot": {
        "newRoot": {"$arrayToObject": "$count_list"}
    }}
]

# 执行聚合查询，游标转列表后取第一条结果（聚合后仅返回1条统计文档）
cursor = restaurants_col.aggregate(pipeline)
count_result = list(cursor)
category_stats = count_result[0] if count_result else {}

# 封装为要求的返回格式，用jsonify自动完成序列化
return jsonify({"restaurants": category_stats})

错误原因说明 #

• CommandCursor是pymongo的游标迭代对象，不是Python原生可序列化结构，所以直接返回会报JSON序列化错误，必须先遍历转成列表/字典类型
• 未做结果合并的聚合查询会返回每个分类一条文档的数组，需要在聚合管道里加合并步骤，才能得到键为分类名、值为计数的单个对象

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