python 用 pymysql 执行数据库通用查询

一、加载库

import pandas as pd import pymysql

二、定义函数创建连接和游标

def mycursor(db_name=None): '''连接数据库，创建游标''' config = dict(zip(['host', 'user', 'port', 'password'], ['192.168.137.155', 'shanger', 3306, '0123'])) config.update(database=db_name) connection = pymysql.connect(**config) cursor = connection.cursor() return cursor

调用函数

 

 

三、定义通用查询函数

def query(sql): '''以数据框形式返回查询据结果''' cursor.execute(sql) data = cursor.fetchall()  # 以元组形式返回查询数据
    header = [t[0] for t in cursor.description] df = pd.DataFrame(list(data), columns=header)  # pd.DataFrem 对列表具有更好的兼容性
    return df

 

四、调用函数执行查询

1、用户查询

sql = 'select user, host from mysql.user;' query(sql)

 结果

 

  2、数据库查询

sql = 'show databases;' query(sql)

结果