pandas.DataFrame.to_sql方法如何指定primary key並且修改dataframe的列

解決辦法如下:

  1. 創建一個表,說明數據
# name和id同時作爲primary key

create table grade(
	name varchar(20),
	id int,
	sex varchar(2),
	primary key(name,id)
)
import pandas as pd
data = {
	'mingzi':['john','lily'],
	'ID':[10,20]
	'xingbie':['male','female']
}
df = pd.DataFrame(data)
  1. 修改dataframe的column名
# 插入這個dataframe到grade表中,顯然表結構和dataframe的column不同,必須修改它
df.columns = ['name','id','sex']
  1. 插入dataframe到表中
from sqlalchemy import create_engine

engine = create_engine('sqlite:///e:\\class.db')

data.to_sql('grade', engine, index=False, if_exists='append') #append或者replace

大功告成

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章