Python 数据处理扩展包： pandas 模块的DataFrame介绍（读写数据库的操作）

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不能存在