需求
最近公司干活,收到一个需求,说是让手动将数据库查出来的信息复制粘贴到excel中,在用excel中写好的公式将指定的两列数据用update这样的语句替换掉。
例如:
有个A库,其中有两个A.01和A.02字段,需要将这两个字段替换到下面的sql语句中,
1
|
update A set A. 01 = 'excel第一列的值' where A. 02 = 'excel第二列的值'
|
虽然excel中公式写好了,但是还需要将总计的那行复制粘贴到txt文档中,所以索性太麻烦,果断用Python写了一个自动化的脚本,虽然查出的数据还需要手动复制excel中,但是后续操作即可用脚本完成
读取excel指定列实现生成txt
废话不多说直接上代码:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
#coding=utf-8
from openpyxl import load_workbook
#第一套servicebus.t_spis_servicebus_svcmain表的
#读取excel的数据
def read_excel():
#打开一个workbook
wb = load_workbook(filename = 'xxx.xlsx' )
#获取所有表格(worksheet)的名字
sheets = wb.get_sheet_names()
#遍历每一个sheet,并且拿到worksheet对象
for i in range ( len (sheets)):
ws = wb.get_sheet_by_name(sheets[i])
#声明list
list_A_B = []
#遍历A,B列的所有数据值,ws.max_row:获取最大行数
for rowNum in range ( 1 ,ws.max_row + 1 ):
if (ws.cell(row = rowNum,column = 1 ).value ! = None ) and (ws.cell(row = rowNum,column = 2 ).value ! = None ):
#ws.cell(row=rowNum,column=1).value:拿到指定列的行数据
sql = 'update A set A.01=' + "'" + ws.cell(row=rowNum,column=1).value + "' " + 'where A.02=' + "'" + ws.cell(row = rowNum,column = 2 ).value + "'" + ';'
list_A_B.append(sql)
#print(list_A_B)
text_save(list_A_B, 'A.txt' )
print ( str (sheets[i]) + '已完成!' )
print ( '执行成功!' )
#将读取的数据存入txt文档中
def text_save(content,filename,mode = 'a' ):
#打开文件
file = open (filename,mode)
for i in range ( len (content)):
number = content[i] + '\n'
file .write(number)
file .close()
if __name__ = = '__main__' :
read_excel()
|
总结
其中用到的是openpyxl这个库,个人感觉非常好用,初步写的时候参考了一下国外大佬的一篇博客,附上链接:
以上这篇Python读取excel指定列生成指定sql脚本的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/s740556472/article/details/77666895