Python讀取excel指定列生成指定sql腳本的方法

更新時間：2018年11月28日 10:31:10 作者：sy_y

今天小編就為大家分享一篇Python讀取excel指定列生成指定sql腳本的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

需求

最近公司干活，收到一個需求，說是讓手動將數據庫查出來的信息復制粘貼到excel中，在用excel中寫好的公式將指定的兩列數據用update這樣的語句替換掉。

例如：

有個A庫，其中有兩個A.01和A.02字段，需要將這兩個字段替換到下面的sql語句中，

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

雖然excel中公式寫好了，但是還需要將總計的那行復制粘貼到txt文檔中，所以索性太麻煩，果斷用Python寫了一個自動化的腳本，雖然查出的數據還需要手動復制excel中，但是后續(xù)操作即可用腳本完成

讀取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('執(zhí)行成功！')

#將讀取的數據存入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腳本的方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: