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