python fetchall()转化为数据框_python连接mysql数据库获取数据并转化为dataframe

import pandas as pd

import pymysql #该库用于python和mysql的连接

#参考：https://www.runoob.com/python3/python3-mysql.html

#打开数据库连接，db为数据库名称

db = pymysql.connect(host="localhost",user="root",passwd="China110@",db="practice" )

#定义获取函数

def get_df_from_db(sql):

cursor = db.cursor()#使用cursor()方法获取用于执行SQL语句的游标

cursor.execute(sql)# 执行SQL语句

"""

使用fetchall函数以元组形式返回所有查询结果并打印出来

fetchone()返回第一行，fetchmany(n)返回前n行

游标执行一次后则定位在当前操作行，下一次操作从当前操作行开始

"""

data = cursor.fetchall()

#下面为将获取的数据转化为dataframe格式

columnDes = cursor.description #获取连接对象的描述信息

columnNames = [columnDes[i][0] for i in range(len(columnDes))] #获取列名

df = pd.DataFrame([list(i) for i in data],columns=columnNames) #得到的data为二维元组，逐行取出，转化为列表，再转化为df

"""

使用完成之后需关闭游标和数据库连接，减少资源占用,cursor.close(),db.close()

db.commit()若对数据库进行了修改，需进行提交之后再关闭

"""

cursor.close()

db.close()

print("cursor.description中的内容：",columnDes)

return df

sql = "SELECT * FROM score" # SQL语句

dff=get_df_from_db(sql)

dff

以上，之后就可以进行进一步的分析可视化了。