pymysql获取要查询的字段名(列名)

使用pymysql连接数据库进行查询时,获取的只是查询的结果,并不包含列名。
可以使用cursor.description来获取列名的相关信息。执行结果如下所示。

#!/usr/bin/env/python
# -*- coding:utf-8 -*-
import pymysql
import pandas as pd
import time
def 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.description
    result = cursor.fetchall()
#执行结果转化为dataframe
    df = pd.DataFrame(list(result))
# 关闭连接
    db.close()
#返回dataframe
    return result,col

sql1 = """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', '预警类型编码', '预警说明', '指标率', '阈值说明']
  • 11
    点赞
  • 30
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值