import pymysql

# 連接 MySQL 數(shù)據(jù)庫
conn = pymysql.connect(
    host='159.xxx.xxx.216',  # 主機名
    port=3306,         # 端口號，MySQL默認為3306
    user='xxxx',       # 用戶名
    password='xxxx', # 密碼
    database='xx',   # 數(shù)據(jù)庫名稱
)

# 創(chuàng)建游標對象
cursor = conn.cursor()
# 執(zhí)行 SQL 查詢語句
cursor.execute("SELECT * FROM users WHERE gender='female'")
# 獲取查詢結果
result = cursor.fetchall()

import pandas as pd
# 將查詢結果轉化為 Pandas dataframe 對象
df = pd.DataFrame(result, columns=[i[0] for i in cursor.description])

# 關閉游標和數(shù)據(jù)庫連接
cursor.close()
conn.close()

# 步驟 1：創(chuàng)建 MySQL 數(shù)據(jù)庫連接引擎
from sqlalchemy import create_engine
# 創(chuàng)建 MySQL 數(shù)據(jù)庫連接引擎
engine = create_engine('mysql+pymysql://username:password@host:port/database')
步驟 2：執(zhí)行 SQL 查詢語句并將結果轉化為 Pandas dataframe 對象
import pandas as pd
# 執(zhí)行 SQL 查詢語句，并將結果轉化為 Pandas dataframe 對象
df = pd.read_sql("SELECT * FROM users WHERE gender='female'", con=engine)
# 關閉數(shù)據(jù)庫連接
engine.dispose()

import pandas as pd
import pymysql
from sqlalchemy import create_engine
def query_mysql(sql_query, host=None, port=None, user=None, password=None, database=None, engine=None):
    """
    連接 MySQL 數(shù)據(jù)庫，執(zhí)行查詢，并將查詢結果轉化為 Pandas DataFrame 對象。
    :param sql_query: SQL 查詢語句
    :param host: 主機名，默認為 None
    :param port: 端口號，默認為 None
    :param user: 用戶名，默認為 None
    :param password: 密碼，默認為 None
    :param database: 數(shù)據(jù)庫名稱，默認為 None
    :param engine: SQLAlchemy 的數(shù)據(jù)庫引擎對象，默認為 None
    :return: Pandas DataFrame 對象
    """
    # 如果未提供數(shù)據(jù)庫連接引擎，則使用 pymysql 庫連接 MySQL 數(shù)據(jù)庫
    if engine is None:
        # 連接 MySQL 數(shù)據(jù)庫
        conn = pymysql.connect(
            host=host,
            port=port,
            user=user,
            password=password,
            database=database,
        )
        # 創(chuàng)建游標對象
        cursor = conn.cursor()
        # 執(zhí)行 SQL 查詢語句
        cursor.execute(sql_query)
        # 獲取查詢結果
        result = cursor.fetchall()
        # 將查詢結果轉化為 Pandas DataFrame 對象
        df = pd.DataFrame(result, columns=[i[0] for i in cursor.description])
        # 關閉游標和數(shù)據(jù)庫連接
        cursor.close()
        conn.close()
    # 如果已提供數(shù)據(jù)庫連接引擎，則使用 SQLAlchemy 庫連接 MySQL 數(shù)據(jù)庫
    else:
        # 執(zhí)行 SQL 查詢語句，并將結果轉化為 Pandas DataFrame 對象
        df = pd.read_sql(sql_query, con=engine)
    return df

# 使用 pymysql 庫連接 MySQL 數(shù)據(jù)庫
df1 = query_mysql(
    sql_query="SELECT * FROM users WHERE gender='female'",
    host='159.xxx.xxx.216',  # 主機名
    port=3306,         # 端口號，MySQL默認為3306
    user='xxxx',       # 用戶名
    password='xxxx', # 密碼
    database='xx',   # 數(shù)據(jù)庫名稱
)
# 使用 SQLAlchemy 庫連接 MySQL 數(shù)據(jù)庫
engine = create_engine('mysql+pymysql://xxx:xxx@localhost:3306/ad')
df2 = query_mysql(sql_query="SELECT * FROM users WHERE gender='female'", engine=engine)

# 法一:
import pymysql
import pandas as pd
def query_data(sql_query):
    # 連接數(shù)據(jù)庫
    conn = pymysql.connect(
        host='xxx.xxx.xxx.xxx',  # 主機名
        port=3306,         # 端口號，MySQL默認為3306
        user='xxx',       # 用戶名
        password='xxx', # 密碼
        database='xxx',   # 數(shù)據(jù)庫名稱
    )
    try:
        # 創(chuàng)建游標對象
        cursor = conn.cursor()
        # 執(zhí)行 SQL 查詢語句
        cursor.execute(sql_query)
        # 獲取查詢結果
        result = cursor.fetchall()
        # 獲取查詢結果的字段名和元數(shù)據(jù)
        columns = [col[0] for col in cursor.description]
        # 將查詢結果封裝到 Pandas DataFrame 中
        df = pd.DataFrame(result, columns=columns)
        return df
    finally:
        # 關閉游標和連接
        cursor.close()
        conn.close()
db_data = query_data(sql_query)
# 法二:
from sqlalchemy import create_engine
import pandas as pd
def getdata_from_db(query, db, host='xxx.xxx.xxx.xxx', port=3306, user='xxx', password='xxx'):
    try:
        engine = create_engine(f'mysql+pymysql://{user}:{password}@{host}:{port}/{db}?charset=utf8')
        # 使用 with 語句自動管理連接的生命周期
        with engine.connect() as conn:
            data = pd.read_sql(query, conn)
        return data
    except Exception as e:
        print(f"Error occurred when executing SQL query: {e}")
        return None
db_data = getdata_from_db(sql_query, 'ad')
# 法三：超級精簡版
from sqlalchemy import create_engine
import pandas as pd
engine = create_engine(f'mysql+pymysql://xxx:xxx@xxx:3306/xx?charset=utf8')
db_data = pd.read_sql(sql, engine)
db_data.head()