使用 Python 脚本将数据导入 MySQL 数据库

1. 简介

本技术文档介绍如何使用 Python 脚本将 CSV 数据导入 MySQL 数据库。这份指南将涵盖以下步骤：

• 设置 MySQL 连接
• 读取 CSV 文件
• 将数据插入数据库
• 错误处理

2. 环境配置

• Python: 确保已安装 Python 3.x 版本。
• MySQL Connector/Python: 使用 pip install mysql-connector-python 安装 MySQL Connector/Python 库。
• CSV 文件: 准备包含要导入数据的 CSV 文件。
• MySQL 数据库: 创建一个新的数据库和表格来存储数据。

3. 代码示例

```python import mysql.connector import csv

数据库连接配置

config = { 'user': 'your_username', 'password': 'your_password', 'host': 'localhost', 'database': 'your_database_name' }

CSV 文件路径

csv_file_path = 'your_csv_file.csv'

try: # 创建数据库连接 conn = mysql.connector.connect(**config)

# 创建游标
cursor = conn.cursor()

# 读取 CSV 文件
with open(csv_file_path, 'r') as file:
    reader = csv.DictReader(file)

    # 获取 CSV 文件的列名
    column_names = reader.fieldnames

    # 生成 INSERT 语句
    insert_query = f"INSERT INTO your_table_name ({','.join(column_names)}) VALUES ({','.join(['%s']*len(column_names))})"

    # 逐行插入数据
    for row in reader:
        # 将值转换为列表
        values = list(row.values())
        cursor.execute(insert_query, values)

# 提交更改
conn.commit()
print("数据已成功导入数据库！")

except mysql.connector.Error as error: print("错误:", error)

finally: # 关闭连接 if conn: cursor.close() conn.close() ```

4. 代码说明

• mysql.connector 库用于连接 MySQL 数据库。
• csv 库用于读取 CSV 文件。
• 代码首先使用 mysql.connector.connect() 函数连接到数据库。
• 然后创建游标对象，用于执行 SQL 语句。
• csv.DictReader 函数用于读取 CSV 文件，并将每一行作为字典返回。
• insert_query 变量包含 INSERT 语句，使用 %s 占位符表示要插入的值。
• 循环遍历 CSV 文件的每一行，将数据插入数据库。
• 最后，使用 conn.commit() 提交更改，并将数据库连接关闭。

5. 注意事项

• 确保 CSV 文件中的列名与数据库表中的列名一致。
• 可以根据需要修改代码中的数据库连接配置、CSV 文件路径和表格名称。
• 在插入数据之前，建议先验证 CSV 文件中的数据格式，并进行必要的预处理。

6. 结论

本技术文档详细介绍了使用 Python 脚本将数据导入 MySQL 数据库的方法。该方法简单易行，且可用于处理各种格式的数据。在实际应用中，可以根据具体需求修改代码以满足不同的导入场景。