MongoDB中如何为数组指定位置元素添加字段并修改属性

问题场景 #

现有如下MongoDB文档：

from pymongo import MongoClient

client = MongoClient(some_mongodb_url)
coll = client.my_database.my_collection

# 查询到的文档
{
    '_id': 1,
    'items_list': [
        {
            'item_id': 'one',
            'size': 1
        },
        {
            'item_id': 'two',
            'size': 2,
            'color': 'blue'
        }
    ]
}

需要更新该文档，为items_list数组的第0个元素添加color: 'red'，并将其size修改为2。

原代码的问题 #

你尝试的代码使用$addFields操作符后，每个数组元素都被添加了名为0的字段，而非修改目标元素。原因是在聚合管道中，items_list.0的写法会被解析为给数组内每个元素添加字段，而非定位数组的第0个元素。

正确解法 #

方法1：用$set直接定位字段（推荐，适合修改部分字段） #

通过点语法直接指定数组索引下的具体字段，用$set精准更新：

coll.update_one(
    {'_id': 1},
    {
        '$set': {
            'items_list.0.size': 2,
            'items_list.0.color': 'red'
        }
    },
    upsert=False
)

方法2：用$replaceAt替换整个元素（适合MongoDB 4.4+，完全替换元素） #

如果需要完全替换数组指定位置的元素，可使用$replaceAt操作符：

updated_item = {
    'item_id': 'one',
    'size': 2,
    'color': 'red'
}

coll.update_one(
    {'_id': 1},
    [
        {
            '$set': {
                'items_list': {
                    '$replaceAt': {
                        'input': '$items_list',
                        'index': 0,
                        'newValue': updated_item
                    }
                }
            }
        }
    ],
    upsert=False
)

验证结果 #

执行上述任一代码后，查询文档即可得到期望结果：

coll.find_one({'_id': 1})
# 返回结果
{
    '_id': 1,
    'items_list': [
        {
            'item_id': 'one',
            'size': 2,
            'color': 'red'
        },
        {
            'item_id': 'two',
            'size': 2,
            'color': 'blue'
        }
    ]
}

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