PyMongo查询MongoDB如何按指定ID列表顺序返回匹配结果

你之前的聚合查询失效的核心原因是$indexOfArray对比时类型不匹配：你传入的ID列表是字符串格式，而MongoDB中文档的_id是ObjectId类型，类型不一致导致无法匹配索引，所有__order字段都被赋值为-1，排序自然不会符合预期。

方案1：调整聚合查询（兼容所有MongoDB版本） #

第一步先将你的字符串ID列表统一转为ObjectId类型数组，再传入聚合逻辑：

from bson.objectid import ObjectId

# 你的原始顺序ID列表（字符串格式）
raw_id_list = ["id8", "id5", "id4", "id9", "id3"]
# 转换为ObjectId类型数组
oid_list = [ObjectId(_id) for _id in raw_id_list]

# 执行聚合查询
result = db.collection.aggregate([
    # 匹配所有目标ID的文档
    {"$match": {"_id": {"$in": oid_list}}},
    # 基于转换后的ObjectId数组计算排序权重
    {"$addFields": {"__order": {"$indexOfArray": [oid_list, "$_id"]}}},
    # 按自定义顺序排序
    {"$sort": {"__order": 1}},
    # 可选：删除临时排序字段，不影响返回结果结构
    {"$unset": "__order"}
])

方案2：MongoDB 5.2+ 简化写法 #

如果你使用的MongoDB版本≥5.2，可以直接用内置的$sortByArray算子，无需额外计算排序字段：

result = db.collection.aggregate([
    {"$match": {"_id": {"$in": oid_list}}},
    {"$sortByArray": {"field": "$_id", "sortBy": oid_list}}
])

方案3：小数据量下Python层面排序 #

如果查询的文档数量不多，可以在业务代码层面完成排序，逻辑更简单易维护：

# 先查询所有匹配的文档，构建_id为键的查询字典
doc_map = {str(doc["_id"]): doc for doc in db.collection.find({"_id": {"$in": oid_list}})}
# 按原始ID列表顺序重组结果，自动过滤不存在的ID
sorted_docs = [doc_map[_id] for _id in raw_id_list if _id in doc_map]

如果你的ID列表中存在集合里没有的无效ID，$indexOfArray会返回-1，这类文档会排在结果最前面。如果需要过滤无效ID，可以在$addFields阶段后新增{"$match": {"__order": {"$ne": -1}}}逻辑。

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