1. 程式人生 > >python中讀寫excel

python中讀寫excel

一 讀取excel

這裡介紹一個不錯的包xlrs,可以工作在任何平臺。這也就意味著你可以在Linux下讀取Excel檔案。

首先,開啟workbook;    
import xlrd
wb = xlrd.open_workbook('myworkbook.xls')

檢查表單名字:    
wb.sheet_names()

得到第一張表單,兩種方式:索引和名字    
sh = wb.sheet_by_index(0)
sh = wb.sheet_by_name(u'Sheet1')

遞迴打印出每行的資訊:    
for rownum in range(sh.nrows):
    print sh.row_values(rownum)

如果只想返回第一列資料:
first_column = sh.col_values(0)

通過索引讀取資料:    
cell_A1 =  sh.cell(0,0).value
cell_C4 = sh.cell(rowx=3,colx=2).value

注意:這裡的索引都是從0開始的。

二 寫excel 

這裡介紹一個不錯的包xlwt,可以工作在任何平臺。這也就意味著你可以在Linux下儲存Excel檔案。 基本部分

在寫入Excel表格之前,你必須初始化workbook物件,然後新增一個workbook物件。比如:
import xlwt
wbk = xlwt.Workbook()
sheet = wbk.add_sheet('sheet 1')

這樣表單就被建立了,寫入資料也很簡單:
# indexing is zero based, row then column
sheet.write(0,1,'test text')

之後,就可以儲存檔案(這裡不需要想開啟檔案一樣需要close檔案):
wbk.save('test.xls')

深入探索

worksheet物件,當你更改表單內容的時候,會有警告提示。
sheet.write(0,0,'test')
sheet.write(0,0,'oops')
 
# returns error:
# Exception: Attempt to overwrite cell:
# sheetname=u'sheet 1' rowx=0 colx=0

解決方式:使用cell_overwrite_ok=True來建立worksheet:
sheet2 =  wbk.add_sheet('sheet 2', cell_overwrite_ok=True)
sheet2.write(0,0,'some text')
sheet2.write(0,0,'this should overwrite')

這樣你就可以更改表單2的內容了。

更多
# Initialize a style
style = xlwt.XFStyle()
 
# Create a font to use with the style
font = xlwt.Font()
font.name = 'Times New Roman'
font.bold = True
 
# Set the style's font to this new one you set up
style.font = font
 
# Use the style when writing
sheet.write(0, 0, 'some bold Times text', style)

xlwt 允許你每個格子或者整行地設定格式。還可以允許你新增連結以及公式。其實你可以閱讀原始碼,那裡有很多例子:

    dates.py, 展示如何設定不同的資料格式
    hyperlinks.py, 展示如何建立超連結 (hint: you need to use a formula)
    merged.py, 展示如何合併格子
    row_styles.py, 展示如何應用Style到整行格子中.

傳說中python操作ms office功能最強大的是win32com,但只能要ms上使用。

不過對於比較簡單的需求顯得有些小題大作。那麼來看下簡單的,分別是xlrd和xlwt模組,

不過暫時只支援excel2003 ,但是可以跨平臺操作。

xlrd

http://pypi.python.org/pypi/xlrd
簡單使用
匯入 import xlrd 開啟excel data = xlrd.open_workbook('demo.xls') #注意這裡的workbook首字母是小寫 檢視檔案中包含sheet的名稱 data.sheet_names() 得到第一個工作表,或者通過索引順序 或 工作表名稱 table = data.sheets()[0] table = data.sheet_by_index(0)
table = data.sheet_by_name(u'Sheet1') 獲取行數和列數 nrows = table.nrows ncols = table.ncols 獲取整行和整列的值(陣列) table.row_values(i) table.col_values(i) 迴圈行,得到索引的列表 for rownum in range(table.nrows): print table.row_values(rownum) 單元格 cell_A1 = table.cell(0,0).value cell_C4 = table.cell(2,3).value 分別使用行列索引 cell_A1 = table.row(0)[0].value cell_A2 = table.col(1)[0].value 簡單的寫入 row = 0 col = 0 ctype = 1 # 型別 0 empty,1 string, 2 number, 3 date, 4 boolean, 5 error value = 'lixiaoluo' xf = 0 # 擴充套件的格式化 (預設是0) table.put_cell(row, col, ctype, value, xf) table.cell(0,0) # 文字:u'lixiaoluo' table.cell(0,0).value # 'lixiaoluo'

xlwt

http://pypi.python.org/pypi/xlrd

簡單使用

匯入xlwt

import xlwt

新建一個excel檔案

file = xlwt.Workbook() #注意這裡的Workbook首字母是大寫,無語吧

新建一個sheet

table = file.add_sheet('sheet name')

寫入資料table.write(行,列,value)

table.write(0,0,'test')

如果對一個單元格重複操作,會引發

returns error:
# Exception: Attempt to overwrite cell:
# sheetname=u'sheet 1' rowx=0 colx=0

所以在開啟時加cell_overwrite_ok=True 解決

table = file.add_sheet('sheet name',cell_overwrite_ok=True )

儲存檔案

file.save('demo.xls')

另外,使用style

style = xlwt.XFStyle() # 初始化樣式

font = xlwt.Font() #為樣式建立字型

font.name = 'Times New Roman'

font.bold = True

style.font = font #為樣式設定字型

table.write(0, 0, 'some bold Times text', style) # 使用樣式

xlwt 允許單元格或者整行地設定格式。還可以新增連結以及公式。可以閱讀原始碼,那裡有例子:

dates.py, 展示如何設定不同的資料格式

hyperlinks.py, 展示如何建立超連結 (hint: you need to use a formula)

merged.py, 展示如何合併格子

row_styles.py, 展示如何應用Style到整行格子中.

具體的例子可以看:

google論壇:


三 修改excel 

Python中一般使用xlrd(excel read)來讀取Excel檔案,使用xlwt(excel write)來生成Excel檔案(可以控制Excel中單元格的格式),需要注意的是,用xlrd讀 取excel是不能對其進行操作的:xlrd.open_workbook()方法返回xlrd.Book型別,是隻讀的,不能對其進行操作。而 xlwt.Workbook()返回的xlwt.Workbook型別的save(filepath)方法可以儲存excel檔案。

因此對於讀取和生成Excel檔案都非常容易處理,但是對於已經存在的Excel檔案進行修改就比較麻煩了。不過,還有一個xlutils(依賴於xlrd和xlwt)提供複製excel檔案內容和修改檔案的功能。其實際也只是在xlrd.Book和xlwt.Workbook之間建立了一個管道而已,如下圖:

 

xlutils.copy模組的copy()方法實現了這個功能,示例程式碼如下:

from xlrd import open_workbook
from xlutils.copy import copy
 
rb = open_workbook('m:\\1.xls')
 
#通過sheet_by_index()獲取的sheet沒有write()方法
rs = rb.sheet_by_index(0)
 
wb = copy(rb)
 
#通過get_sheet()獲取的sheet有write()方法
ws = wb.get_sheet(0)
ws.write(0, 0, 'changed!')
 
wb.save('m:\\1.xls')

四 參考 

http://pypi.python.org/pypi/xlrd http://pypi.python.org/pypi/xlwt http://pypi.python.org/pypi/xlutils

本文整合自 

http://www.leyond.info/write-excel-files-with-python-using-xlwt/ http://www.leyond.info/read-excel-from-python-using-xlrs/ http://www.zhlwish.com/2010/10/09/python_edit_excel/

完!