通过匹配_id更新MongoDB集合

Question

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

就像我在python数据帧中找到_id的匹配项一样，我需要在集合中更新相应的文档。下面的脚本正在工作，但是如果文档太多，那么执行任何有效的方法都需要执行时间。 请指教

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()

Answer 1

您可以尝试使用批量写入。 您只需要创建一个包含所有更新的数组，然后使用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/