PyMongo find的$in查询如何动态传入数组替代硬编码枚举

使用PyMongo操作MongoDB数据库做数据检索时，需要动态传入长度不固定的数组作为查询条件，不想硬编码枚举数组下标。核心需求是通过循环类写法，在查询条件中动态表示[postLike[0], postLike[1] ...]这类长度可变的数组元素序列。

当前可正常运行但存在硬编码问题的代码如下：

@app.route("/user/addlikevideo", methods=["POST","GET"])
@jwt_required()
def add_like_video():
   if request.method =="GET":
        try:
            getTokenEmail = get_jwt()["sub"] # 从Token中获取邮箱
            existing_user = user_collection.find_one({"email": getTokenEmail}) # 查询对应用户
            postLike = existing_user["likeVideos"] # 拿到格式类似[4,25]的点赞视频ID数组

            # 需要替换硬编码[postLike[0],postLike[1]]的动态写法
            thePostLike = post_collection.find({ "idCount": { "$in": [postLike[0],postLike[1]]}})
            listThePostLike = list(thePostLike)
            
            json_data = dumps(listThePostLike, indent = 4) 
            return json_data

        except:
            return "no liked post"

完全不需要手动循环展开数组下标，MongoDB的$in运算符原生支持直接接收任意长度的列表作为匹配值集合，直接把取到的postLike数组整体传入即可，不管数组长度是0、几十还是上百，都能正常完成匹配。
如果需要对数组元素做预处理（比如类型转换、过滤空值/无效值），用Python的列表推导（对应你提到的for循环、map逻辑）处理完再传入就行，根本不需要逐个枚举数组下标。

修改后的可动态适配任意长度数组的代码如下：

@app.route("/user/addlikevideo", methods=["POST","GET"])
@jwt_required()
def add_like_video():
   if request.method =="GET":
        try:
            getTokenEmail = get_jwt()["sub"]
            existing_user = user_collection.find_one({"email": getTokenEmail})
            # 兼容用户不存在、点赞字段不存在的场景，默认赋值空数组
            postLike = existing_user.get("likeVideos", []) if existing_user else []
            
            # 空数组直接返回结果，避免无效数据库查询
            if not postLike:
                return dumps([], indent=4)

            # 如需预处理元素（比如转整型、过滤无效值），用列表推导即可，示例：
            # processed_like_ids = [int(pid) for pid in postLike if pid is not None]
            # 直接把整个数组传给$in，无需手动展开下标
            thePostLike = post_collection.find({ "idCount": { "$in": postLike }})
            listThePostLike = list(thePostLike)
            
            return dumps(listThePostLike, indent = 4)

        except Exception as e:
            return "no liked post"

补充说明 #

• 如果你需要的不是直接传入原数组，而是要对每个元素做查询逻辑加工（比如每个元素对应一个$or条件），同样可以用列表推导先构造好查询条件列表，再直接传入对应运算符即可，不需要硬编码每一项。
• 不建议写裸except，最好捕获具体异常，避免掩盖真实代码报错原因，方便排查问题。

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