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)

結果