使用pymysql连接数据库进行查询时,获取的只是查询的结果,并不包含列名。
可以使用cursor.description来获取列名的相关信息。执行结果如下所示。
#!/usr/bin/env/python# -*- coding:utf-8 -*-import pymysqlimport pandas as pdimport timedef execude_sql(sql):# 创建连接try:db = pymysql.connect(host='127.0.0.1', port=3308, user='name', passwd='password', db='db', charset='utf8')except:print('数据库连接失败,10s后重试')time.sleep(10)# 创建游标cursor = db.cursor()cursor.execute(sql)col = cursor.descriptionresult = cursor.fetchall()#执行结果转化为dataframedf = pd.DataFrame(list(result))# 关闭连接db.close()#返回dataframereturn result,colsql1 = """select comment_sql from etl_event_head where event_id=6001"""resu,co = execude_sql(sql1)for (test,) in resu:result,col = execude_sql(test)print(result)print(col[1][0])print(col)
执行结果:
(('40102', 'cp_tt2fre', '联系人→进件', '6%', '低于80%'), ('4105', 'cp_orders', '单卡→生成订单', '3%', '低于5%'))预警类型编码(('ERROR_CODE', 253, None, 92, 92, 0, True), ('预警类型编码', 253, None, 144, 144, 0, False), ('预警说明', 253, None, 192, 192, 0, True), ('指标率', 253, None, 76, 76, 0, True), ('阈值说明', 253, None, 192, 192, 0, True))
如果想获取字段列表,遍历一下元组即可。
list = []for i in range(len(col)):list.append(col[i][0])print(list)
输出
['ERROR_CODE', '预警类型编码', '预警说明', '指标率', '阈值说明']