MongoDB聚合阶段如何定义变量？Python聚合管道函数报错求助

你遇到的问题核心是混淆了Python变量和MongoDB聚合管道内的变量：$$criteria 是用来引用MongoDB聚合上下文里定义的变量（比如通过$let、管道阶段变量或聚合选项里的变量），但你的criteria是Python函数的参数，不属于MongoDB的变量上下文，所以会报错"未定义的变量"。

下面给你两种最实用的解决方案，根据你的需求场景选择：

方案1：直接动态注入字段名（适用于按单个字段分组的场景） #

如果你的criteria是要分组的字段名（比如"country"、"status"），直接在Python层面把字段名转换成MongoDB的字段引用格式$字段名即可：

from typing import List

def categorize_by(criteria) -> List:
    return [
        {
            "$group": {
                "_id": f"${criteria}",  # 用Python的f-string动态生成MongoDB字段引用
                "count": {"$sum": 1},
                "projects": {
                    "$push": {
                        "_id": "$_id",
                        "name": "$name",
                    },
                },
            },
        },
    ]

调用示例：比如按"country"分组，直接传categorize_by("country")，生成的管道里_id就是"$country"，完全符合MongoDB的语法要求。

方案2：传入完整的MongoDB表达式（适用于复杂分组逻辑） #

如果你的分组条件是更复杂的逻辑（比如按年龄区间、多字段组合、条件判断），可以让criteria直接是一个MongoDB的表达式对象，然后直接赋值给_id：

from typing import List

def categorize_by(criteria) -> List:
    return [
        {
            "$group": {
                "_id": criteria,  # 直接传入MongoDB表达式
                "count": {"$sum": 1},
                "projects": {
                    "$push": {
                        "_id": "$_id",
                        "name": "$name",
                    },
                },
            },
        },
    ]

调用示例：比如按是否成年分组，先定义MongoDB的条件表达式，再传入函数：

age_criteria = {
    "$cond": {
        "if": {"$gte": ["$age", 18]},
        "then": "adult",
        "else": "minor"
    }
}
pipeline = categorize_by(age_criteria)

为什么原来的写法不行？ #

MongoDB的$$var语法是用来引用聚合管道内部定义的变量，比如用$let定义的变量：

{
  "$let": {
    "vars": {"groupKey": "$country"},
    "in": "$$groupKey"
  }
}

但你的criteria是Python层面的变量，不属于MongoDB的聚合上下文，所以不能用$$引用。直接在Python中动态构建管道是更直接且符合你需求的方式。

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