pandas.DataFrame.to_sql的进度条

from tqdm import tqdm
import pandas as pd
from sqlalchemy import create_engine

# 创建数据库连接
engine = create_engine('mysql://username:password@localhost/mydatabase')

# 读取数据到DataFrame
df = pd.read_csv('data.csv')

# 获取DataFrame的行数
total_rows = len(df)

# 创建进度条
with tqdm(total=total_rows, desc='Writing to SQL') as pbar:
    # 将DataFrame数据写入SQL数据库
    df.to_sql('mytable', con=engine, if_exists='replace', index=False, chunksize=1000)
    pbar.update(total_rows)  # 更新进度条到100%