要將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數據庫中。
注意:根據您的需求和數據量,您可能需要調整腳本中的錯誤處理、事務控制和性能優化。