Python程式設計：pandas讀寫mysql資料

安裝用到的子產品

pip install pandas sqlalchemy pymysql      

1、建表

# 建表
create table student(
id int(11) primary key auto_increment,
name varchar(10) default "",
age int(11) default 0
)      

如果不建立表也可以，會自動建立，不過一些字段可能不是你想要的，如果明确字段還需要自己提前建表

2、字典資料讀寫

# -*- coding: utf-8 -*-

# 導入必要子產品
import pandas as pd
from sqlalchemy import create_engine

# 連接配接資料庫
engine = create_engine('mysql+pymysql://root:123456@localhost:3306/demo?charset=utf8')


# 寫入資料
data = [
    {'id': 1, 'name': '小明', 'age': 23},
    {'id': 2, 'name': '小紅', 'age': 24},
    {'id': 3, 'name': '小白', 'age': 25}
]

df = pd.DataFrame(data)

df.to_sql('student', engine, if_exists="append", index=False)


# 讀取資料
sql = 'select * from student;'

df = pd.read_sql_query(sql, engine)

print(df.to_dict(orient="records"))
"""
[{'id': 1, 'name': '小明', 'age': 23}, 
{'id': 2, 'name': '小紅', 'age': 24}, 
{'id': 3, 'name': '小白', 'age': 25}]
"""

      

3、csv檔案讀寫

資料 student.csv

id,name,age
1,小明,23
2,小紅,24
3,小白,25
      

可以不建立資料表，會自動建立

import pandas as pd
from sqlalchemy import create_engine

engine = create_engine('mysql+pymysql://root:123456@localhost:3306/demo?charset=utf8')

# 讀取檔案導入資料庫，表不存在會建立
df = pd.read_csv("student.csv", sep=',')

df.to_sql('student', engine, index=False)

# 從資料庫讀取資料
df = pd.read_sql_query("select * from student", engine)
print(df)
"""
   id name  age
0   1   小明   23
1   2   小紅   24
2   3   小白   25
"""      

參考

用Pandas庫實作MySQL資料庫的讀寫