python连接mysql_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

python连接mysql_python连接mysql数据库获取数据并转化为dataframe_第1张图片

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

你可能感兴趣的:(python连接mysql)