pandas 生成 SQL

时间：2023-05-23 09:15:28 标签：pandas python sql

看过来

《pandas 教程》持续更新中，可作为 pandas 入门进阶课程、pandas 中文手册、用法大全，配有案例讲解和速查手册。提供建议、纠错、催更等加作者微信: gairuo123（备注：pandas教程）和关注公众号「盖若」ID: gairuo。查看更新日志。作者开办 Python 数据分析培训，详情 Python 数据分析培训。

本教程作者所著新书《深入浅出Pandas：利用Python进行数据处理与分析》（ISBN：9787111685456）已由机械工业出版社出版上市，各大电商平台有售，欢迎：查看详情并关注购买。

现有以下 2020 年节假日的数据，需要将其插入数据库的 holiday 表里，holiday 除了以下三列，还有一个年份字段 year。

'''
节日 开始日期 结束日期
元旦 2020-01-01 2020-01-01
除夕 2020-01-24 2020-01-24
清明节 2020-04-04 2020-04-04
劳动节 2020-05-01 2020-05-01
端午节 2020-06-25 2020-06-25
国庆节 2020-10-01 2020-10-01
'''

读取数据：

import pandas as pd

df = pd.read_clipboard()
df.head()
'''
节日 开始日期 结束日期
0 元旦 2020-01-01 2020-01-01
1 除夕 2020-01-24 2020-01-24
2 清明节 2020-04-04 2020-04-04
3 劳动节 2020-05-01 2020-05-01
4 端午节 2020-06-25 2020-06-25
'''

然后对 DataFrame 进行迭代，生成 insert SQL 语句：

sql = ''
for i,r in df.iterrows():
    r_sql = f"INSERT INTO `holiday` (`holiday`, `year`, `start_date`, `end_date`) 
        VALUES ('{r['节日']}', '{r['结束日期'][:4]}', '{r['开始日期']}', '{r['结束日期']}');"
    sql = sql + r_sql + '\n'
print(sql)
'''
INSERT INTO `holiday` (`holiday`, `year`, `start_date`, `end_date`) VALUES 
    ('元旦', '2020', '2020-01-01', '2020-01-01');
INSERT INTO `holiday` (`holiday`, `year`, `start_date`, `end_date`) VALUES 
    ('除夕', '2020', '2020-01-24', '2020-01-24');
INSERT INTO `holiday` (`holiday`, `year`, `start_date`, `end_date`) VALUES 
    ('清明节', '2020', '2020-04-04', '2020-04-04');
INSERT INTO `holiday` (`holiday`, `year`, `start_date`, `end_date`) VALUES 
    ('劳动节', '2020', '2020-05-01', '2020-05-01');
INSERT INTO `holiday` (`holiday`, `year`, `start_date`, `end_date`) VALUES 
    ('端午节', '2020', '2020-06-25', '2020-06-25');
INSERT INTO `holiday` (`holiday`, `year`, `start_date`, `end_date`) VALUES 
    ('国庆节', '2020', '2020-10-01', '2020-10-01');
'''

生成后，可以复制这些 SQL 语句，执行。

（完）

注：此案例收录在《深入浅出Pandas：利用Python进行数据处理与分析》17.2.8 小节。

pandas 生成 SQL

相关内容