CSV文件是指逗號分隔值文件,它是一種常用的簡單的數據交換格式。而MySQL則是一種常用的關系型數據庫管理系統。在實際的應用場景中,我們經常需要將CSV文件的數據導入到MySQL中進行進一步的處理。下面就為大家介紹一下如何使用Python實現將CSV文件導入到MySQL數據庫中。
首先需要安裝好Python的MySQL數據庫驅動程序(比如PyMySQL),以及將CSV文件讀取到Python中的Pandas模塊。接下來,我們可以使用pandas.read_csv方法讀取CSV文件數據,并將其轉換成pandas.DataFrame對象。其中,我們需要注意設置CSV文件編碼方式、列名等參數。
import pandas as pd file_path = 'data.csv' df = pd.read_csv(file_path, encoding='utf-8', header=None, names=['id', 'name', 'age'])
接下來,我們可以使用pymysql庫連接MySQL數據庫,創建表格,并將數據插入到表格中。
import pymysql conn = pymysql.connect(host='localhost', port=3306, user='root', password='password', db='test') cursor = conn.cursor() # 創建表格 create_table_sql = 'CREATE TABLE IF NOT EXISTS `user` (`id` int NOT NULL, `name` varchar(20) NOT NULL, `age` int NOT NULL, PRIMARY KEY(`id`))' cursor.execute(create_table_sql) for index, row in df.iterrows(): insert_sql = 'INSERT INTO `user` (`id`, `name`, `age`) VALUES ("%d", "%s", "%d")' % (row['id'], row['name'], row['age']) cursor.execute(insert_sql) conn.commit() cursor.close() conn.close()
最后,我們可以使用SELECT語句查詢表格中的數據,以確保數據已經成功導入到MySQL數據庫中。
# 查詢數據 select_sql = 'SELECT * FROM `user`' result = cursor.execute(select_sql) print(result.fetchall())
通過以上代碼,我們可以輕松地將CSV文件中的數據導入到MySQL數據庫中,從而方便我們進行數據的管理和處理。