大数跨境
0
0

如何用Python快速生成SQL语句?比偷懒,我是第一名!

如何用Python快速生成SQL语句?比偷懒,我是第一名! 数据分析与统计学之美
2022-01-17
2

大家好,我是黄同学!

作为一名搞数据的,写SQL是每天必不可少的工作。而我又是一个喜欢偷懒的人,就想着能不能使用Python快速生成SQL语句呢?

假如现在想建一个临时表自己练习,我使用create table建表后,还需要使用多条INSERT INTO语句向表中插入数据,具体语法如下图所示。

写入一行数据还好,但是如果N多行,手动就略微麻烦了。

我之前要么在编辑器Sublime Text3里多行操作,要么在Excel文件中使用函数生成SQL插入语句。

▲Sublime Text3

如果想进一步偷懒,可以怎么做呢?

我希望能够使用Python一键将csv文件中的数据转成SQL语句,并输出到剪贴板上,这样我们直接粘贴到SQL编辑器上就可以直接运行了!

所要用到的表格数据(data_1.csv)如下所示:

我们可以这样做,Pandas模块读取csv文件的数据后,使用for循环依次读取每一行的数据。并通过f-string格式化字符串,一键生成我们所需的SQL语句。

import pandas as pd
df = pd.read_csv("data_1.csv")
tables = "mysql.newtable"
text = ""
for index, row in df.iterrows():
    text = f"{text}INSERT INTO {tables} VALUES('{row[0]}','{row[1]}','{row[2]}','{row[3]}','{row[4]}');\n"
text

在字符串中我们增加了"\n"用来换行,运行结果如下所示。

那么如何将输出的结果写入剪贴板呢?

import pyperclip
pyperclip.copy(text)

这个pyperclip模块贼简单,就两个函数:

  • copy()用于向剪贴板发送文本;
  • paste()用于从剪贴板接收文本。

如果我们将两段代码一起运行,就会在鼠标的剪切板上出现SQL语句,具体效果如下所示。

▲运行动图GIF

这样我只需运行该Python代码,再粘贴到SQL编辑器中。

执行一下,成功插入数据。

其实这种方法还可以应用在很多类似的场景,例如UPDATE语句、DELETE语句等等,思路都是一致的。

推荐阅读    点击标题可跳转
0. 深度对比Python4大文件/文件夹处理库!
1. Python快速实现分列转到行!
2. 对比学习12款Python数据可视化库!
3. 如何用Python分析股票收益率?
4. pprint漂亮的打印,强烈推荐大家学习这个库!
5. 数据分析方法论,大总结!
6. 有了这款Python神器,新手也会调试代码!
7. Pandas与openpyxl库的 "完美" 融合!
8. Python提取PDF简历中的信息,写入Excel
9. Linux系列:完美收官,一共22个课时!
10. Python操作MySQL数据库!
11. 一个「神奇」的Python库,99%的人都爱!
12. 对比Excel/MySQL/Python,讲述if的使用原理!
13. 坐地铁,就能学会的3种"非常有趣"的Python玩法!-1
14. 坐地铁,就能学会的3种"非常有趣"的Python玩法!-2
15. 坐地铁,就能学会的3种"非常有趣"的Python玩法!-3
【声明】内容源于网络
0
0
数据分析与统计学之美
免费领10w字"Python知识手册",共400页,后台回复“十万”领取!
内容 1080
粉丝 0
数据分析与统计学之美 免费领10w字"Python知识手册",共400页,后台回复“十万”领取!
总阅读76
粉丝0
内容1.1k