Question

python 数据库查询到的结果如何能赋值给dataframe并标明列明

用下面语句取出了数据库中的数据，怎样才能将他们赋值给DataFrame,并按SQL中的字段名给DataFrame的数据标明列名，谢谢
count=cur.execute('select bug_type,bug_severity,count(bug_type),ft_id from tm_bug group by ft_id,bug_type,bug_severity')
result=cur.fetchall()

Answer 1

cursor.execute(sql)
columns_name = [tuple[0] for tuple in cursor.description]
data= pd.DataFrame(cursor.fetchall(), columns=columns_name)
cursor.close()
这样写

追问

2个问题,
为什么第二句要用tuple[0]
第三句输出data是报错没有被定义data is not defined

追答

第一个问题
你print下cursor.description就知道里面是什么结构了
第二个问题
data是被赋值的，怎么可能报未定义呢？你有没有装pandas啊？

追问

第一个问题已经明白，谢谢

第二个问题,我的确装了pandas并且导入了上图附件是我的代码，请问有什么问题吗？谢谢

另外用fetchall或fetchone取出来的数据是每条（行）数据包括四个字段的四列数据么（四列分别是bug_type,bug_severity,count(bug_type,ft_id),感谢

追答

我写的是oracle的。你的是mysql。
出错原因在于mysql查询回来的数据的结构是元组嵌套的，pandas.DataFrame识别不了.。所以要把查询回来的数据改成dict结构，这样就能识别了。就加一个游标的参数。
cursor = db.cursor(mysql.cursors.DictCursor)
写个示例代码，以后照着写
import pymysql as mysql
import pandas as pd
db = mysql.connect(host="10.2.2.2",user="AIRPORT_DATA",password="157626",database="AIRPORT_DATA" )
cursor = db.cursor(mysql.cursors.DictCursor)
sql = "select * from DATA_TABLE"
cursor.execute(sql)
columns_name = [tuple[0] for tuple in cursor.description]
data= pd.DataFrame(cursor.fetchall(), columns=columns_name)
cursor.close()
db.close()
fetchall或fetchone取出来的数据，有哪些字段，是你的sql决定的。你的sql查了哪些就是哪些。