PyMongo 3.6.1中aggregate(pipeline)无结果问题求助

嘿，我一眼就揪出了导致聚合返回空游标的两个关键问题，咱们逐个来解决：

1. distinct 调用的语法错误 #

你当前的distinct代码里，字段名没有用字符串包裹：

# 错误写法（字段名未加引号，会被识别为未定义变量）
db.ratings.distinct(movieId, {userId: 1})

在PyMongo中，字段名必须以字符串形式传入，修正后应该是：

# 正确写法
movies_rated_by_user = db.ratings.distinct("movieId", {"userId": 1})

2. 聚合管道里的字段名拼写错误 #

你的$match阶段中，"_id "后面多了一个空格！Movies集合的主键字段是_id（无空格），这个多余的空格会让MongoDB找不到匹配的文档，直接返回空结果：

# 错误写法（_id后多了空格）
{"$match": {"_id ": {"$in": movies_rated_by_user}}},

去掉空格即可修复：

# 正确写法
{"$match": {"_id": {"$in": movies_rated_by_user}}},

修正后的完整代码 #

把这两处错误修复后，你的代码就能正常输出预期结果了：

# 获取用户1评分过的电影ID列表
movies_rated_by_user = db.ratings.distinct("movieId", {"userId": 1})

# 修正后的聚合管道
pipeline = [
    {"$match": {"_id": {"$in": movies_rated_by_user}}},
    {"$unwind": "$genres"},
    {"$group": {"_id": "$genres", "movies": {"$addToSet": "$_id"}}}
]

# 执行聚合并打印结果
result = db.movies.aggregate(pipeline)
for doc in result:
    print(doc)

更高效的优化方案（一次聚合完成） #

其实你可以省去先调用distinct的步骤，直接通过一次聚合关联Ratings集合，减少一次数据库查询，效率更高：

pipeline = [
    # 关联Ratings集合，匹配用户1的评分记录
    {"$lookup": {
        "from": "ratings",
        "localField": "_id",
        "foreignField": "movieId",
        "as": "user_ratings"
    }},
    # 只保留用户1评分过的电影
    {"$match": {"user_ratings.userId": 1}},
    # 展开电影类型数组
    {"$unwind": "$genres"},
    # 按类型分组，收集对应的movieId
    {"$group": {"_id": "$genres", "movies": {"$addToSet": "$_id"}}},
    # 可选：按类型名称排序，让结果更规整
    {"$sort": {"_id": 1}}
]

result = db.movies.aggregate(pipeline)
for doc in result:
    print(doc)

这个优化后的管道会直接从Movies集合出发，一步完成关联过滤、类型拆分和分组统计，输出你想要的结果。

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