Python 数据处理扩展包: pandas 模块的DataFrame介绍(读写数据库的操作)

时间:2023-02-02 21:08:56

1、读取表中的内容,如下例子:

import MySQLdb
try:
conn = MySQLdb.connect(host='127.0.0.1',user='root',passwd='root',db='mydb',port=3306)
df = pd.read_sql('select * from test;', con=conn)
conn.close()
print "finish load db"
except MySQLdb.Error,e:
print e.args[1]

注意,上面read_sql方法返回的数据的类型是DataFrame

2、写入数据到表中,如下例子

df = pd.DataFrame([[1,'xxx'],[2,'yyy']],columns=list('ab'))

try:
conn = MySQLdb.connect(host='127.0.0.1',user='root',passwd='root',db='mydb',port=3306)
df.to_sql('test1',conn,flavor='mysql')
conn.close()
print "finish load db"
except MySQLdb.Error,e:
print e.args[1]

说明:上面代码要求表test1不能存在