學了python這么久,總算是在工作中用到一次。這次是為了從excel文件中讀取數據然后寫入到數據庫中。這個邏輯用java來寫的話就太重了,所以這次考慮通過python腳本來實現。
在此之前需要給python添加一個xlrd模塊,這個模塊是專門用來操作excel文件的。
在mac中可以通過easy_install xlrd命令實現自動安裝模塊
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
|
import xdrlib ,sys import xlrd def open_excel( file = a.xlsx'): try : data = xlrd.open_workbook( file ) #打開excel文件 return data except Exception,e: print str (e) def excel_table_bycol( file = 'a.xlsx' ,colindex = [ 0 ],table_name = 'Sheet1' ): data = open_excel( file ) table = data.sheet_by_name(table_name) #獲取excel里面的某一頁 nrows = table.nrows #獲取行數 colnames = table.row_values( 0 ) #獲取第一行的值,作為key來使用,對于不同的excel文件可以進行調整 list = [] #(1,nrows)表示取第一行以后的行,因為第一行往往是表頭 for rownum in range ( 1 ,nrows): row = table.row_values(rownum) if row: app = {} for i in colindex: app[ str (colnames[i]).encode( "utf-8" )] = str (row[i]).encode( "utf-8" ) #將數據填入一個字典中,同時對數據進行utf-8轉碼,因為有些數據是unicode編碼的 list .append(app) #將字典加入列表中去 return list def main(): #colindex是一個數組,用來選擇讀取哪一列,因為往往excel中的一小部分才是我們需要的 tables = excel_table_bycol(colindex = [ 1 , 4 ],table_name = u 'areaCode' ) file = open ( 'channel_area_code.sql' , 'w' ) #創建sql文件,并開啟寫模式 for row in tables: if row[ 'area_code' ] ! = '': file .write( "update table_name set para1='%s' where para2='%s';\n" % (row[ 'para1' ],row[ 'para2' ])) #往文件里寫入sql語句 if __name__ = = "__main__" : main() |
這并非是一個通用的python腳本,還是需要根據excel文件的格式作出一些調整,但是代碼并不復雜,開發速度也很快,比以前用java是輕松多了。
感謝閱讀,希望能幫助到大家,謝謝大家對本站的支持!
原文鏈接:https://my.oschina.net/u/241670/blog/349239