pandas 读写mysql数据库 csv文件

一、读取mysql数据

#方式一
import pymysql
import pandas as pd

pd.set_option('display.max_columns', None)  #显示所有列
pd.set_option('display.max_rows', None)  #显示所有行
con = pymysql.connect(host,user,password,db=db,charset='utf8')  #连接数据库
sql = "SELECT * FROM tablename"
df = pd.read_sql(sql,con=con)  #读取数据
print(df)

#方式二 使用sqlalchemy
from sqlalchemy import create_engine
import pandas as pd

con = create_engine('mysql+pymysql://user:password@host:3306/db_name?charset=utf8')  ## 初始化数据库链接
sql = "SELECT * FROM sw_area"
df = pd.read_sql(sql,con=con)   #读取数据
print(df)

 

二、写入mysql数据

import pandas as pd
from sqlalchemy import create_engine

# pandans 不能使用pymysql库插入数据

a =['1','2','3']#列表a
b = ['4','5','6']#列表b
df = pd.DataFrame({'a':a, 'b':b})
# 初始化数据库连接
con = create_engine('mysql+pymysql://user:password@host:3306/db_name?charset=utf8')
#if_exists='append' 如果表存在则添加数据
# pd.io.sql.to_sql(df,'tablename',con=con,if_exists='append',index=None)
# or
df.to_sql('tablename',con=con,if_exists='append',index=None)

三、读写CSV、EXCEL文件

import pandas as pd

#读取csv文件
read = pd.read_csv('data.csv')
print(read)

#写入csv文件
df = pd.DataFrame({'a':[1,2,3],'b':[2,3,4]})
df.to_csv('data.csv')


#读取excel文件
read = pd.read_excel('data.csv')
print(read)

#写入excel文件
df = pd.DataFrame({'a':[1,2,3],'b':[2,3,4]})
df.to_excel('text.xls',sheet_name='sheet1',index=None)