python - 使用 PyMongo 将 Pandas 数据框插入 mongodb

Question

使用 PyMongo 将 pandas DataFrame 插入 mongodb 的最快方法是什么？ ?

尝试

db.myCollection.insert(df.to_dict())

报错

InvalidDocument: documents must have only string keys, the key was
Timestamp('2013-11-23 13:31:00', tz=None)

---

 db.myCollection.insert(df.to_json())

报错

TypeError: 'str' object does not support item assignment

---

 db.myCollection.insert({id: df.to_json()})

报错

InvalidDocument: documents must have only string a keys, key was <built-in function id>

---

df

<class 'pandas.core.frame.DataFrame'>
DatetimeIndex: 150 entries, 2013-11-23 13:31:26 to 2013-11-23 13:24:07
Data columns (total 3 columns):
amount    150  non-null values
price     150  non-null values
tid       150  non-null values
dtypes: float64(2), int64(1)

Answer 1

这里有最快捷的方法。使用 insert_many来自 pymongo 3 的方法和 to_dict 的“记录”参数方法。

db.collection.insert_many(df.to_dict('records'))