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数据库的读写