2

我正在尝试使用 _id 上的条件从 Python 更新 mongodb 集合

就像我在 python 数据框中找到了 _id 的匹配项,我需要更新 colllection 中的相应文档 下面的脚本正在工作,但是如果文档太多,则执行需要时间,是否有任何有效的方法来处理这个问题。请指教

for document in db.AMTicketData.find():
    for index, row in AMTicketData1.iterrows():
        if(row['_id']==a['_id']):
            db.AMTicketData.update_one({'_id': row['_id']},{'$set': {'Application_Name': row['Application_Name']}}, upsert=True)
            break

我使用了以下批量操作代码,能够批量更新集合

bulk = db.AMTicketData.initialize_unordered_bulk_op()
for index, row in AMTicketData1.iterrows():
    bulk.find({'_id':row['_id']}).update_one({'$set':{'Application_Name':row['Application_Name']}})

bulk.execute()
4

1 回答 1

2

您可以尝试使用批量写入。您只需创建一个包含所有更新的数组并使用 collection.bulk_write(list_of_updates) 应用一次

就像是:

updates = []
for document in db.AMTicketData.find():
   for index, row in AMTicketData1.iterrows():
     if(row['_id']==a['_id']):
      updates.append(UpdateOne({'_id': row['_id']}, {'$set': {'Application_Name': row['Application_Name']}}, upsert=True)
      break

db.AMTicketData.bulk_write(updates)

https://docs.mongodb.com/manual/core/bulk-write-operations/

于 2019-03-20T19:49:15.620 回答