使用PyMongo将Pandas DataFrame写入MongoDB集合

时间:2018-10-24 12:49:38

标签: mongodb pandas dataframe pymongo

我正在尝试使用Python读取csv文件,将其转换为Pandas Dataframe并将数据写入MongoDB中的集合中。以下是我正在尝试的源代码,它运行时出现0个错误,但我无法在试图获取记录的MongoDB集合上看到预期的结果。有什么建议吗?

import pandas as pd
import pymongo
import json


def import_content(filepath):
    mng_client = pymongo.MongoClient('localhost', 27017)
    mng_db = mng_client['PLM']  # mongo db name
    collection_name = 'Sample'  # mongo db collection name
    db_cm = mng_db[collection_name]
    import os
    cdir = os.path.dirname(__file__)
    file_res = os.path.join(cdir, filepath)

    data = pd.read_csv(file_res)
    data_json = json.loads(data.to_json(orient='values'))
    db_cm.remove()
    db_cm.insert(data_json)


if __name__ == " __main__":
    filepath = 'C:/MongoDB/DataSheets/InputFilecsv.csv'  # pass csv file path
    import_content(filepath)

进程结束,退出代码为0

注意:我已经在MongoDB中如下创建了集合名称“ Sample”

{
    "_id" : ObjectId("5bd048d21e36618d869f5e5e"),
    "specid" : "",
    "specname" : "",
    "shortname" : "",
    "status" : "",
    "wf" : "",
    "cat" : "",
    "subcat" : "",
    "grp" : ""
}

0 个答案:

没有答案