MongoDB如何过滤数组内符合条件元素并仅返回指定字段

你之前的两种查询写法存在两个核心问题：

• 普通find方法的投影仅能控制顶层/嵌套字段的显隐，无法实现数组内元素的条件过滤，更不能对过滤后的数组元素做字段裁剪
• 聚合管道中仅使用$filter只能筛选出符合条件的数组元素，不会自动剔除元素内的冗余字段；同时初始$match直接对比$items.x和$items.y是数组整体的顺序比较，存在边界场景判断不准的问题。

正确实现代码 #

使用$filter筛选符合条件的数组元素后，搭配$map遍历裁剪每个元素，仅保留需要的x/y字段即可，代码兼容MongoDB 3.4及以上版本：

fx = mycol.aggregate([
    # 匹配items数组中至少存在一个x不等于y的元素的文档
    {"$match": {
        "items": {
            "$elemMatch": {
                "$expr": {"$ne": ["$x", "$y"]}
            }
        }
    }},
    # 投影返回字段，同时完成数组过滤+元素字段裁剪
    {"$project": {
        "_id": 0,
        "SomeText": 1,
        "items": {
            "$map": {
                # 先过滤出x≠y的有效数组元素，作为map遍历的输入
                "input": {
                    "$filter": {
                        "input": "$items",
                        "as": "item",
                        "cond": {"$ne": ["$$item.x", "$$item.y"]}
                    }
                },
                "as": "validItem",
                # 每个元素仅保留x、y字段，剔除status、price等冗余内容
                "in": {
                    "x": "$$validItem.x",
                    "y": "$$validItem.y"
                }
            }
        }
    }},
    {"$limit": 5}
])

for res in fx:
    print(res)

执行结果 #

运行后将完全返回预期结构：

{'SomeText': 'aaa', 'items': [{'x': 1, 'y': None}, {'x': 2, 'y': None}]}
{'SomeText': 'bbb', 'items': [{'x': 1, 'y': 2}]}

ccc文档因为items内唯一元素x、y均为1，会被$match阶段过滤，不会出现在结果中。

逻辑说明 #

• $match阶段使用$elemMatch逐元素判断数组内是否存在满足x≠y的记录，完全贴合筛选需求，避免数组整体顺序比较带来的边界错误。
• $project阶段先通过$filter剔除数组内x==y的无效元素，再通过$map遍历剩余有效元素，逐一生成仅包含x、y字段的新对象，从根源上去除冗余字段。

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