当前位置： 移动技术网 > IT编程>脚本编程>Python > Python读取excel指定列生成指定sql脚本的方法

Python读取excel指定列生成指定sql脚本的方法

2018年12月04日  | 移动技术网IT编程  | 我要评论

山神佳誉,杭州 桑拿 小姐,叶嘉莉

需求

最近公司干活，收到一个需求，说是让手动将数据库查出来的信息复制粘贴到excel中，在用excel中写好的公式将指定的两列数据用update这样的语句替换掉。

例如：

有个a库，其中有两个a.01和a.02字段，需要将这两个字段替换到下面的sql语句中，

update a set a.01 = 'excel第一列的值' where a.02 = 'excel第二列的值'

虽然excel中公式写好了，但是还需要将总计的那行复制粘贴到txt文档中，所以索性太麻烦，果断用python写了一个自动化的脚本，虽然查出的数据还需要手动复制excel中，但是后续操作即可用脚本完成

读取excel指定列实现生成txt

废话不多说直接上代码：

#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脚本的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持移动技术网。

您可能感兴趣的文章:

如对本文有疑问，请在下面进行留言讨论，广大热心网友会与你互动！！ 点击进行留言回复

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网