Die Methode, die Sie benötigen, ist update_one()
mit upsert=True
in einer Schleife; Sie können insert_many()
nicht verwenden aus zwei Gründen; Erstens fügen Sie nicht immer ein; irgendwann aktualisierst du; zweitens update_many()
(und insert_many()
) funktionieren nur mit einem einzigen Filter; in Ihrem Fall ist jeder Filter anders, da sich jede Aktualisierung auf eine andere Zeit bezieht.
Dies ist eine generische Lösung, die Datenrahmen kombiniert (df_a
, df_b
in diesem Fall - Sie können so viele haben, wie Sie möchten) in der Weise, die Sie benötigen. Es verwendet iterrows
um jede Zeile des Datenrahmens zu erhalten, filtert nach dem Datum und legt die Werte auf die im Datenrahmen fest. das $set
Der Operator überschreibt Werte, wenn sie bereits vorhanden sind, und setzt sie, wenn sie nicht gesetzt sind. upsert=True
führt eine Einfügung durch, wenn es keine Übereinstimmung für das Datum gibt.
for df in [df_a, df_b]:
for _, row in df.iterrows():
db.mycollection.update_one({'date': row.get('date')}, {'$set': row.to_dict()}, upsert=True)
Vollständig ausgearbeitetes Beispiel:
from pymongo import MongoClient
from pprint import pprint
import datetime
import pandas as pd
# Sample data setup
db = MongoClient()['mydatabase']
data_a = [[datetime.datetime(2017, 5, 19, 21, 20), 96, 8, 98],
[datetime.datetime(2017, 5, 19, 21, 21), 95, 8, 97],
[datetime.datetime(2017, 5, 19, 21, 22), 95, 8, 97]]
df_a = pd.DataFrame(data_a, columns=['date', 'std_500_1000window', 'std_50_100window', 'std_50_2000window'])
data_b = [[datetime.datetime(2017, 5, 19, 21, 20), 98, 9, 10],
[datetime.datetime(2017, 5, 19, 21, 21), 98, 9, 10],
[datetime.datetime(2017, 5, 19, 21, 22), 98, 9, 10]]
df_b = pd.DataFrame(data_b, columns=['date', 'std_50_3000window', 'std_50_300window', 'std_50_500window'])
# Perform the upserts
for df in [df_a, df_b]:
for _, row in df.iterrows():
db.mycollection.update_one({'date': row.get('date')}, {'$set': row.to_dict()}, upsert=True)
# Print the results
for record in db.mycollection.find():
pprint(record)
Ergebnis:
{'_id': ObjectId('5f0ae909df5531ac655ce528'),
'date': datetime.datetime(2017, 5, 19, 21, 20),
'std_500_1000window': 96,
'std_50_100window': 8,
'std_50_2000window': 98,
'std_50_3000window': 98,
'std_50_300window': 9,
'std_50_500window': 10}
{'_id': ObjectId('5f0ae909df5531ac655ce52a'),
'date': datetime.datetime(2017, 5, 19, 21, 21),
'std_500_1000window': 95,
'std_50_100window': 8,
'std_50_2000window': 97,
'std_50_3000window': 98,
'std_50_300window': 9,
'std_50_500window': 10}
{'_id': ObjectId('5f0ae909df5531ac655ce52c'),
'date': datetime.datetime(2017, 5, 19, 21, 22),
'std_500_1000window': 95,
'std_50_100window': 8,
'std_50_2000window': 97,
'std_50_3000window': 98,
'std_50_300window': 9,
'std_50_500window': 10}