读取xlsx文件存入MongoDB数据库中

news2024/9/30 9:37:09

要将xlsx文件存入MongoDB数据库，您需要执行以下步骤：

步骤1：安装必要的库

您需要安装pymongo和openpyxl库。您可以使用以下命令安装：

pip install pymongo openpyxl

步骤2：创建一个MongoDB数据库

您需要创建一个MongoDB数据库，以便将数据存储在其中。您可以使用MongoDB Compass或命令行创建数据库。

步骤3：编写Python脚本

以下是一个Python脚本，可将xlsx文件读取为Pandas数据框架，然后将数据插入MongoDB数据库。

在以上代码中，我们首先连接到MongoDB数据库并指定数据库和集合。然后我们加载D盘一个名为“城市对照表.xlsx”的xlsx文件并将其读入一个Pandas数据框架。最后，我们将数据插入MongoDB数据库。

一、代码：

import pymongo
import pandas as pd
from openpyxl import load_workbook

# 连接MongoDB数据库
client = pymongo.MongoClient("mongodb://localhost:27017/")
db = client["pms"]
collection = db["address"]

# 加载xlsx文件
workbook = load_workbook(filename="D:/城市对照表.xlsx")
worksheet = workbook.active
# 读取数据为Pandas数据框架
data = worksheet.values
columns = next(data)
df = pd.DataFrame(data, columns=columns)

# 将数据插入MongoDB数据库
records = df.to_dict('records')
collection.insert_many(records)

print("数据已成功插入MongoDB数据库")

二、代码：

import openpyxl
import pymongo

# 连接MongoDB数据库
client = pymongo.MongoClient("mongodb://localhost:27017/")
db = client["customer_master_data"]
collection = db["base_pms"]

# 打开xlsx文件
wb = openpyxl.load_workbook('D:/城市对照表.xlsx')
sheet = wb.active
data = []
for row in sheet.iter_rows(min_row=2, values_only=True):
    PKID = row[0]
    RegionName = row[1]
    RegionLevel = row[2]
    data.append({"PKID": PKID, "RegionName": RegionName, "RegionLevel": RegionLevel})

collection.insert_one({'InfoTypeID': 80000, 'data': data})

print("数据已成功更新/插入MongoDB数据库")