Python MongoDB：如何高效修改和删除字典中的列表？

首先先明确你提供的原始MongoDB数据结构（需要先导入from bson import ObjectId）：

from bson import ObjectId

original_data = [
    {
        '_id': ObjectId('5ae348f19268ff41a8e35170'),
        'commission': [
            {
                '_id': ObjectId('5ae348f19268ff41a8e35170'),
                'commission': [
                    {
                        '_id': ObjectId('5ae21c969268ff4118df7f8a'),
                        'market': 'amazon_my',
                        'commission': 4.0,
                        'commission_vat': 0.24,
                        'payment': 2
                    },
                    {
                        '_id': ObjectId('5ae21c969268ff4118df7f8b'),
                        'market': 'amazon_sg',
                        'commission': 4.0,
                        'commission_vat': 0.24,
                        'payment': 2
                    }
                ]
            }
        ]
    }
]

一、MongoDB数据的增删改查操作 #

首先确保你已经通过PyMongo连接到MongoDB：

from pymongo import MongoClient

client = MongoClient('mongodb://localhost:27017/')
db = client['your_database_name']
collection = db['your_collection_name']

1. 查找操作 #

• 查找顶层文档：根据顶层_id快速定位

target_doc = collection.find_one({'_id': ObjectId('5ae348f19268ff41a8e35170')})

• 查找嵌套数组中的特定元素：用$elemMatch精准匹配嵌套列表里的条件

# 只返回符合条件的嵌套commission元素
target_elem = collection.find_one(
    {'_id': ObjectId('5ae348f19268ff41a8e35170')},
    {'commission.commission': {'$elemMatch': {'market': 'amazon_my'}}}
)

2. 插入操作 #

给嵌套的commission列表新增元素，推荐用$push操作符，配合位置占位符$或索引定位父元素：

# 方式1：通过索引定位父commission元素
insert_result = collection.update_one(
    {'_id': ObjectId('5ae348f19268ff41a8e35170')},
    {
        '$push': {
            'commission.0.commission': {
                '_id': ObjectId('5ae348f19268ff41a8e35171'),
                'market': 'amazon_us',
                'commission': 5.0,
                'commission_vat': 0.3,
                'payment': 3
            }
        }
    }
)

# 方式2：用$占位符匹配符合条件的父元素（更灵活，无需知道索引）
insert_result = collection.update_one(
    {
        '_id': ObjectId('5ae348f19268ff41a8e35170'),
        'commission._id': ObjectId('5ae348f19268ff41a8e35170')
    },
    {
        '$push': {
            'commission.$.commission': {
                '_id': ObjectId('5ae348f19268ff41a8e35171'),
                'market': 'amazon_us',
                'commission': 5.0,
                'commission_vat': 0.3,
                'payment': 3
            }
        }
    }
)

# 查看插入结果
print(f"匹配文档数: {insert_result.matched_count}, 修改文档数: {insert_result.modified_count}")

3. 删除操作 #

删除嵌套数组中的元素，用$pull操作符移除符合条件的项：

# 删除嵌套列表中market为amazon_sg的元素
delete_result = collection.update_one(
    {
        '_id': ObjectId('5ae348f19268ff41a8e35170'),
        'commission._id': ObjectId('5ae348f19268ff41a8e35170')
    },
    {
        '$pull': {
            'commission.$.commission': {'market': 'amazon_sg'}
        }
    }
)

4. 更新操作 #

修改嵌套数组中的元素，用$set配合array_filters可以精准定位（避免索引依赖）：

# 将amazon_my的commission值改为6.0
update_result = collection.update_one(
    {'_id': ObjectId('5ae348f19268ff41a8e35170')},
    {
        '$set': {
            'commission.$[parent].commission.$[child].commission': 6.0
        }
    },
    array_filters=[
        {'parent._id': ObjectId('5ae348f19268ff41a8e35170')},
        {'child.market': 'amazon_my'}
    ]
)

array_filters可以分别给父、子数组的元素命名，精准匹配需要修改的项，非常适合多层嵌套的场景。

二、Python中高效修改/删除字典中的列表元素 #

当你把MongoDB数据取到Python本地（转为字典）后，处理嵌套列表可以用这些高效方法：

1. 高效修改元素 #

• 列表推导式批量修改：适合批量更新符合条件的元素

# 假设data是从MongoDB取出的单个文档字典
data = original_data[0]

for parent_comm in data['commission']:
    # 把amazon_my的commission改为6.0，其他元素保持不变
    parent_comm['commission'] = [
        {**item, 'commission': 6.0} if item['market'] == 'amazon_my' else item
        for item in parent_comm['commission']
    ]

• 快速定位单个元素修改：用next()生成器找到目标元素，避免遍历整个列表

for parent_comm in data['commission']:
    # 找到第一个market为amazon_my的元素并修改
    target_item = next((item for item in parent_comm['commission'] if item['market'] == 'amazon_my'), None)
    if target_item:
        target_item['commission'] = 6.0
        break  # 找到后立即退出，提升效率

2. 高效删除元素 #

• 列表推导式过滤删除：这是Python中删除列表元素最高效的方式之一（比循环调用remove()快很多）

for parent_comm in data['commission']:
    # 删除所有market为amazon_sg的元素
    parent_comm['commission'] = [item for item in parent_comm['commission'] if item['market'] != 'amazon_sg']

• 按索引删除：如果知道目标元素的索引，用del或pop()

for parent_comm in data['commission']:
    # 删除索引为1的元素（即amazon_sg那条）
    del parent_comm['commission'][1]
    # 如果需要获取被删除的元素，用pop()
    # removed_item = parent_comm['commission'].pop(1)

3. 封装复用逻辑 #

如果需要频繁处理这类嵌套列表，可以把逻辑封装成函数，提升代码可读性和复用性：

def update_commission_by_market(commission_list, target_market, new_commission):
    """根据market更新commission值"""
    return [
        {**item, 'commission': new_commission} if item['market'] == target_market else item
        for item in commission_list
    ]

def delete_commission_by_market(commission_list, target_market):
    """根据market删除元素"""
    return [item for item in commission_list if item['market'] != target_market]

# 使用示例
for parent_comm in data['commission']:
    parent_comm['commission'] = update_commission_by_market(parent_comm['commission'], 'amazon_my', 6.0)
    parent_comm['commission'] = delete_commission_by_market(parent_comm['commission'], 'amazon_sg')

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