python 访问mysql数据库代码封装

python访问mysql数据库,并返回DataFrame类型的数据

import pymysql.cursors
import pandas as pd
import numpy as np

def pyvisitMysql(params ,sql_text):
    ## 数据库参数信息
    conn=pymysql.Connection(host = params.get('ip'), 
                            port = params.get('port'), 
                            user = params.get('username'), 
                            passwd = params.get('passwd'),
                            charset = params.get('charset')
                         ) 
    cursor=conn.cursor()
    cursor.execute(sql_text)
     # 读取字段列名
    index = cursor.description
    row = list()
    for i in range(len(index)):
    	row.append(index[i][0])
      #获取返回信息
    data = cursor.fetchall()
    result = pd.DataFrame(list(data), columns = row)
      #关闭连接,释放资源
    cursor.close()
    conn.close()
    return result 
if __name__=="__main__":
    import pandas as pd
    import numpy as np
    params = {'ip' : ip,
              'port': 端口,  -- 注意端口为数值变量
              'username':用户名,
              'passwd':密码,
			  'charset':字符集
            }
	sql_text = 'select *  from  superhuang_md.att_res_base;'		  
    data_df = pyvisitMysql(params , sql_text)

你可能感兴趣的:(分析工具与数据库交互,数据库,mysql,python)