使用Python脚本将CSV文件的特定列插入MongoDB集合

Question

我有一个Python脚本可以将CSV文件插入MongoDB集合中，

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)

此处的所有CSV文件的列都将插入Mongo Collection中。如何在Mongo Collection中仅插入CSV文件的特定列的用途酶。

我可以对现有代码进行哪些更改。

可以说，我在Mongo中也已经创建了数据库。即使存在数据库，该命令也会起作用（db = client [“数据库名称”]）

Answer 1

您检查过PymongoArrow吗？最新版本具有写支持，您可以将CSV文件导入MongoDB。这是 文档。您也可以使用mongoimport导入CSV文件，文档为，但是我看不到像您使用PymongoArrow那样排除字段的任何方法。