Dataframe数据直接存入mongodb

 

自己记录下，Dataframe数据需要转成字典，并且保证列标题中不能有 “，”等非string字符。

 

import pymongo
import pandas as pd


def main()
　　mongo_client = pymongo.MongoClient('localhost', 27017)
　　db = mongo_client['mydb']
　　mycol = db['your_col']
　　data = your_df.to_dict(orient='record')　
　　mycol.insert_many(data)

if __name__ == '__main__: 
　　main()

 

参考：

https://stackoverflow.com/questions/49221550/pandas-insert-a-dataframe-to-mongodb?r=SearchResults

http://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.to_dict.html