MongoDB聚合：PyMongo实现嵌套对象键合并与值求和问题

错误原因分析 #

你遇到的$arrayToObject requires an object keys of 'k' and 'v'. Found incorrect number of keys:1错误，是因为传递给$arrayToObject的数组元素不符合要求——该操作要求数组中的每个元素必须是包含k（键名）和v（键值）两个字段的对象，而你的管道中生成的数组元素结构不满足这个条件。

正确的聚合管道实现 #

要实现按as_num分组、求和total_signals、合并嵌套对象signals_per_scenario_1m并对对应场景值求和的需求，需要分步骤处理嵌套对象的键值对：

完整的PyMongo聚合代码 #

from pymongo import MongoClient

# 连接MongoDB（根据你的实际配置调整）
client = MongoClient("mongodb://localhost:27017/")
db = client.your_database_name  # 替换为你的数据库名
collection = db.ip_range24_stats

# 定义聚合管道
pipeline = [
    # 第一步：按as_num分组，计算total_signals总和，同时收集所有场景的键值对数组
    {
        "$group": {
            "_id": "$as_num",
            "total_signals": {"$sum": "$total_signals"},
            "scenario_pairs": {
                "$push": {"$objectToArray": "$signals_per_scenario_1m"}
            }
        }
    },
    # 展开二维的场景键值对数组为一维（因为每个文档的场景数组被push后是二维数组）
    {"$unwind": "$scenario_pairs"},
    {"$unwind": "$scenario_pairs"},
    # 第二步：按as_num和场景名称分组，对每个场景的信号值求和
    {
        "$group": {
            "_id": {
                "as_num": "$_id",
                "scenario": "$scenario_pairs.k"
            },
            "total_signals": {"$first": "$total_signals"},
            "sum_scenario_value": {"$sum": "$scenario_pairs.v"}
        }
    },
    # 第三步：再次按as_num分组，重组每个场景的k-v结构数组
    {
        "$group": {
            "_id": "$_id.as_num",
            "total_signals": {"$first": "$total_signals"},
            "scenario_kv_array": {
                "$push": {
                    "k": "$_id.scenario",
                    "v": "$sum_scenario_value"
                }
            }
        }
    },
    # 第四步：将k-v数组转换为嵌套对象，并重命名字段
    {
        "$project": {
            "_id": 0,
            "as_num": "$_id",
            "total_signals": 1,
            "signals_per_scenario_1m": {"$arrayToObject": "$scenario_kv_array"}
        }
    }
]

# 执行聚合并输出结果
for doc in collection.aggregate(pipeline):
    print(doc)

步骤详解 #

1. 第一次$group：按as_num分组，用$sum计算该组的total_signals总和；同时用$objectToArray将每个文档的signals_per_scenario_1m嵌套对象转换为[{k:场景名, v:信号数}, ...]的数组结构，并通过$push收集到scenario_pairs中。
2. 两次$unwind：因为scenario_pairs是二维数组（每个文档贡献一个数组元素），所以需要两次展开将其转为一维数组，方便后续按场景分组求和。
3. 第二次$group：按as_num和场景名称（scenario_pairs.k）分组，用$sum计算该场景在当前as_num下的总信号值，同时保留之前计算的total_signals总和。
4. 第三次$group：再次按as_num分组，将每个场景的名称和求和后的值重新组织为{k:场景名, v:总信号数}的对象，并收集到scenario_kv_array数组中。
5. $project：用$arrayToObject将scenario_kv_array转换为嵌套对象signals_per_scenario_1m，同时重命名字段（去掉_id，将_id改为as_num）。

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