CODEWITHSUNDEEP
pythonpython-3.xmongodbcsvpymongo
Aditya
Aditya

Reputation: 1058

Inserting specific columns of csv file into mongodb collection using python script

I have a python script to insert a csv file into mongodb collection

import pymongo
import pandas as pd
import json

client = pymongo.MongoClient("mongodb://localhost:27017")

df = pd.read_csv("iris.csv")

data = df.to_dict(oreint = "records")

db = client["Database name"]

db.CollectionName.insert_many(data)

Here all the columns of csv files are getting inserted into mongo collection. How can I achieve a usecase where I want to insert only specific columns of csv file in the mongo collection .

What changes I can make to existing code.

Lets say I also have database already created in my Mongo. Will this command work even if the database is present (db = client["Database name"])

Upvotes: 0

Views: 714

Answers (1)

Rachelle Palmer Frick
Rachelle Palmer Frick

Reputation: 71

Have you checked out pymongoarrow? the latest release has write support where you can import a csv file into mongodb. Here are the release notes and documentation. You can also use mongoimport to import a csv file, documentation is here, but I can't see any way to exclude fields like the way you can with pymongoarrow.

Upvotes: 1

Related Questions