要将CSV文件导入MySQL数据库,您可以按照以下步骤操作:
确保您已经安装了MySQL数据库和Python环境。
打开命令行或终端,输入以下命令以安装mysql-connector-python
库(如果尚未安装):
pip install mysql-connector-python
data.csv
,并确保其内容格式正确。例如:id,name,age
1,Alice,30
2,Bob,25
3,Charlie,22
import csv
import mysql.connector
from mysql.connector import Error
# 连接到MySQL数据库
try:
connection = mysql.connector.connect(
host='localhost',
user='your_username',
password='your_password',
database='your_database'
)
if connection.is_connected():
cursor = connection.cursor()
# 删除表(如果存在)
cursor.execute("DROP TABLE IF EXISTS your_table")
# 创建新表
create_table_query = """CREATE TABLE your_table (
id INT PRIMARY KEY,
name VARCHAR(255),
age INT
);"""
cursor.execute(create_table_query)
# 读取CSV文件并插入数据
with open('data.csv', mode='r') as csvfile:
csv_reader = csv.DictReader(csvfile)
insert_query = "INSERT INTO your_table (id, name, age) VALUES (%s, %s, %s)"
cursor.executemany(insert_query, csv_reader)
# 提交更改并关闭连接
connection.commit()
print("CSV data has been successfully imported to MySQL database.")
except Error as e:
print(f"Error while connecting to MySQL: {e}")
finally:
if connection.is_connected():
cursor.close()
connection.close()
print("MySQL connection is closed")
在脚本中填写正确的数据库连接信息(主机名、用户名、密码和数据库名),并根据需要修改表名和字段名。
运行Python脚本,CSV数据将被导入到MySQL数据库中。
注意:根据您的需求和数据量,您可能需要调整脚本中的错误处理、事务控制和性能优化。