安装用到的模块
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
"""