如何使用PyMongo删除MongoDB文档数组中的单个元素实例

问题场景 #

现有MongoDB文档：

{
"_id" : ObjectId("5ec55af811ac5e2e2aafb2b9"),
"username" : "TestUsername",
"tags" : ["gaming", "gaming", "music"]
}

使用$pull操作会删除所有匹配的"gaming"元素：

mycol.update_one({"username": "TestUsername"}, {"$pull": {"tags": "gaming" }})

执行后tags变为["music"]，需求是仅删除一个"gaming"实例。

可行方案 #

方案1：使用$pullOne（MongoDB 4.2+） #

MongoDB 4.2及以上版本提供了$pullOne操作符，专门用于删除数组中第一个匹配指定值的元素，PyMongo中直接调用即可：

mycol.update_one(
    {"username": "TestUsername"},
    {"$pullOne": {"tags": "gaming"}}
)

执行后文档的tags数组会变为["gaming", "music"]，仅移除第一个"gaming"。

方案2：兼容低版本MongoDB #

如果你的MongoDB版本低于4.2，无法使用$pullOne，可以通过定位元素索引+$unset+$pull的组合实现：

# 查找目标文档并获取第一个匹配元素的索引
target_doc = mycol.find_one({"username": "TestUsername", "tags": "gaming"})
if target_doc:
    try:
        tag_index = target_doc["tags"].index("gaming")
        # 先将目标位置元素置为null，再移除数组中的null值
        mycol.update_one(
            {"username": "TestUsername"},
            {
                "$unset": {f"tags.{tag_index}": ""},
                "$pull": {"tags": None}
            }
        )
    except ValueError:
        # 数组中无匹配元素时的处理
        pass

这种方法先定位到第一个"gaming"的位置，将其设为null后，再从数组中移除所有null值，最终只删除单个目标元素。

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