大家好,我是我是菜鸟哥!
作为一名搞数据的,写SQL是每天必不可少的工作。而我又是一个喜欢偷懒的人,就想着能不能使用Python快速生成SQL语句呢?
假如现在想建一个临时表自己练习,我使用create table
建表后,还需要使用多条INSERT INTO
语句向表中插入数据,具体语法如下图所示。
![bf6ea16fc26bfcc07255be9f73093783.png](https://img-blog.csdnimg.cn/img_convert/bf6ea16fc26bfcc07255be9f73093783.png)
写入一行数据还好,但是如果N多行,手动就略微麻烦了。
我之前要么在编辑器Sublime Text3里多行操作,要么在Excel文件中使用函数生成SQL插入语句。
![262ca6d6fee7d7cf9877086c69becba4.png](https://img-blog.csdnimg.cn/img_convert/262ca6d6fee7d7cf9877086c69becba4.png)
如果想进一步偷懒,可以怎么做呢?
我希望能够使用Python一键将csv文件中的数据转成SQL语句,并输出到剪贴板上,这样我们直接粘贴到SQL编辑器上就可以直接运行了!
所要用到的表格数据(data_1.csv)如下所示:
![64bc6649b85b836737307b0ecf963cc8.png](https://img-blog.csdnimg.cn/img_convert/64bc6649b85b836737307b0ecf963cc8.png)
我们可以这样做,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"
用来换行,运行结果如下所示。
![ee09b00e19822225be9565c222d46317.png](https://img-blog.csdnimg.cn/img_convert/ee09b00e19822225be9565c222d46317.png)
那么如何将输出的结果写入剪贴板呢?
import pyperclip
pyperclip.copy(text)
这个pyperclip模块贼简单,就两个函数:
copy()
用于向剪贴板发送文本;paste()
用于从剪贴板接收文本。
如果我们将两段代码一起运行,就会在鼠标的剪切板上出现SQL语句,具体效果如下所示。
![4016adc351a6e2d8fe4d2f8cc4f4fbcf.gif](https://img-blog.csdnimg.cn/img_convert/4016adc351a6e2d8fe4d2f8cc4f4fbcf.gif)
这样我只需运行该Python代码,再粘贴到SQL编辑器中。
![20a47dbde40b32d7b470535d4f32a2db.png](https://img-blog.csdnimg.cn/img_convert/20a47dbde40b32d7b470535d4f32a2db.png)
执行一下,成功插入数据。
![2ce611537874bf155372d524ffeda2dd.png](https://img-blog.csdnimg.cn/img_convert/2ce611537874bf155372d524ffeda2dd.png)
其实这种方法还可以应用在很多类似的场景,例如UPDATE
语句、DELETE
语句等等,思路都是一致的。
好啦,偷懒小技巧加一~
推荐阅读:
入门: 最全的零基础学Python的问题 | 零基础学了8个月的Python | 实战项目 |学Python就是这条捷径
干货:爬取豆瓣短评,电影《后来的我们》 | 38年NBA最佳球员分析 | 从万众期待到口碑扑街!唐探3令人失望 | 笑看新倚天屠龙记 | 灯谜答题王 |用Python做个海量小姐姐素描图 |碟中谍这么火,我用机器学习做个迷你推荐系统电影
趣味:弹球游戏 | 九宫格 | 漂亮的花 | 两百行Python《天天酷跑》游戏!
AI: 会做诗的机器人 | 给图片上色 | 预测收入 | 碟中谍这么火,我用机器学习做个迷你推荐系统电影
小工具: Pdf转Word,轻松搞定表格和水印! | 一键把html网页保存为pdf!| 再见PDF提取收费! | 用90行代码打造最强PDF转换器,word、PPT、excel、markdown、html一键转换 | 制作一款钉钉低价机票提示器! |60行代码做了一个语音壁纸切换器天天看小姐姐!|
年度爆款文案
点阅读原文,看B站我的视频!