Python中使用pymongo实现对MongoDB数据库的数据批量导入导出

import pymongo

import csv

# 连接MongoDB数据库

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

db = client["mydatabase"]

collection = db["mycollection"]

# 数据导入

def import_data(file_path):

    with open(file_path, 'r') as csvfile:

        reader = csv.DictReader(csvfile)

        for row in reader:

            collection.insert_one(row)

    print("数据导入完成")

import_data("data.csv")

# 数据导出

def export_data(file_path):

    with open(file_path, 'w', newline='') as csvfile:

        writer = csv.DictWriter(csvfile, fieldnames=["name", "age", "email"])

        writer.writeheader()

        for document in collection.find():

            writer.writerow(document)

    print("数据导出完成")

export_data("data.csv")

# 使用例子

"""

假设我们有一个名为data.csv的CSV文件，其中包含了一些人的信息，包括姓名、年龄和电子邮箱。

我们想要将这些信息导入到MongoDB数据库，并且在需要的时候将其导出为CSV文件。

数据导入的例子：

data.csv文件内容如下：

name,age,email

Alice,25,alice@example.com

Bob,30,bob@example.com

Cindy,35,cindy@example.com

运行import_data("data.csv")，数据将会被导入到MongoDB数据库中的mydatabase数据库的mycollection集合中。

数据导出的例子：

运行export_data("data.csv")，数据将会从MongoDB数据库中的mydatabase数据库的mycollection集合中导出，并保存到data.csv文件中。

"""

这是一个简单的使用pymongo库实现对MongoDB数据库的数据批量导入导出的例子。

在这个例子中，我们使用了csv库来处理CSV文件，并使用pymongo库来连接MongoDB数据库并操作数据。

在数据导入函数import_data中，我们首先打开CSV文件，使用csv.DictReader读取CSV文件的内容并转换为字典。

然后，我们使用collection.insert_one方法将每一行数据插入到MongoDB数据库的mycollection集合中。

在数据导出函数export_data中，我们首先打开CSV文件，使用csv.DictWriter写入CSV文件的头部内容。

然后，我们使用collection.find方法遍历MongoDB数据库中的所有文档，并将每个文档写入到CSV文件中。

最后，我们提供了一个使用例子，演示了如何使用import_data函数将CSV文件导入到MongoDB数据库，

以及如何使用export_data函数将MongoDB数据库中的数据导出为CSV文件。

通过这个例子，你可以学习如何使用pymongo库来批量导入导出MongoDB数据库中的数据，并进行相应的操作。