將CSV文件寫入到MySQL中
以上的例子實現了使用Pandas庫實現MySQL數據庫的讀寫,我們將再介紹一個實例:將CSV文件寫入到MySQL中,示例的example.csv文件如下
示例的Python代碼如下:
1 # -*- coding: utf-8 -*- 2 3 # 導入必要模塊 4 import pandas as pd 5 from sqlalchemy import create_engine 6 7 # 初始化數據庫連接,使用pymysql模塊 8 db_info = {'user': 'root', 9 'password': '123456', 10 'host': 'localhost', 11 'port': 3306, 12 'database': 'test' 13 } 14 15 engine = create_engine('mysql+pymysql://%(user)s:%(password)s@%(host)s:%(port)d/%(database)s?charset=utf8' % db_info, encoding='utf-8') 16 # 直接使用下一種形式也可以 17 # engine = create_engine('mysql+pymysql://root:123456@localhost:3306/test') 18 19 # 讀取本地CSV文件 20 df = pd.read_csv("C:/Users/fuqia/Desktop/example.csv", sep=',') 21 print(df) 22 # 將新建的DataFrame儲存為MySQL中的數據表,不儲存index列(index=False) 23 # if_exists: 24 # 1.fail:如果表存在,啥也不做 25 # 2.replace:如果表存在,刪了表,再建立一個新表,把數據插入 26 # 3.append:如果表存在,把數據插入,如果表不存在創建一個表!! 27 pd.io.sql.to_sql(df, 'example', con=engine, index=False, if_exists='replace') 28 # df.to_sql('example', con=engine, if_exists='replace')這種形式也可以 29 print("Write to MySQL successfully!")
在MySQL中查看example表格
補充:engine.execute(sql)可以直接執行sql語句:
1 from sqlalchemy import create_engine 2 3 4 engine = create_engine('mysql+pymysql://root:123456@localhost:3306/test') 5 sql = "DROP TABLE IF EXISTS example" 6 engine.execute(sql)
如果用pymysql,則必須用cursor,讀者可以對比一下。
1 import pymysql 2 from sqlalchemy import create_engine 3 4 conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='123456', db='test') 5 # engine = create_engine('mysql+pymysql://root:123456@localhost:3306/test') 6 sql = "DROP TABLE IF EXISTS test_input" 7 cursor = conn.cursor() 8 cursor.execute(sql)