1. 程式人生 > >用 Python 讀寫 Excel 表格

用 Python 讀寫 Excel 表格

Python 可以讀寫 Excel 表格嗎?

當然可以。 Python 下有很多類庫可以做到, openpyxl 就是其中的佼佼者。

openpyxl 的設計非常漂亮 ,你一定會喜歡它!不信請往下看:

工作簿

開始 openpyxl 前,無需提前建好工作簿( Workbook )。 只需匯入 Workbook 類,便可在記憶體中建立新工作簿並開始操作:

>>> from openpyxl import Workbook
>>> wb = Workbook()

新建的工作簿預設預先建好一個工作表,通過 active 屬性即可獲取:

>>> ws = wb.active

註解

如果工作簿包含多個工作表,該屬性將返回第一個。

通過 create_sheet 方法,可以建立新的工作表。 建立可以是在後面追加:

>>> ws1 = wb.create_sheet('Mysheet')

或者,在前面插入:

>>> ws1 = wb.create_sheet('Mysheet', 0)

未指定表格名將自動生成,序列形如: SheetSheet1Sheet2 ,以此類推。 當然了,你覺得不合適可以進行修改:

>>> ws.title = 'New Title'

工作表標題標籤背景顏色預設是白色。 用一個 RGB

顏色程式碼設定 sheet_properties.tabColor 屬性即可修改:

>>> ws.sheet_properties.tabColor = "1072BA"

一旦你給工作表命名,便可以通過該名字來定位:

>>> ws3 = wb["New Title"]

通過 sheetnames 屬性,可以取出所有工作表表名:

>>> print(wb.sheetnames)
['Sheet2', 'New Title', 'Sheet1']

當然了,遍歷所有工作表,直接 for-in 更為優雅:

>>> for sheet in wb:
...     print(sheet.title)

使用 copy_worksheet 方法,可在工作簿內拷貝工作表:

>>> source = wb.active
>>> target = wb.copy_worksheet(source)

從檔案載入

如果已有工作簿,可通過 openpyxl.load_workbook 函式進行載入:

>>> from openpyxl import load_workbook
>>> wb2 = load_workbook('test.xlsx')
>>> print(wb2.sheetnames)
['Sheet2', 'New Title', 'Sheet1']

資料處理

單個單元格

操作工作表,從修改單元格內容開始。單元格可以通過工作表鍵直接訪問:

>>> cell = ws['A4']

這個語句將返回 A4 單元格,或者在單元格不存在時建立它。可以直接賦值:

>>> ws['A4'] = 10

另一種方式是使用 cell 方法訪問單元格,指定行和列:

>>> cell = ws.cell(row=4, column=2, value=10)

註解

工作表建立後,不包含任何單元格,單元格在第一次被訪問時自動建立。

多單元格

連續多個單元格可以通過切片獲得:

>>> cell_range = ws['A1':'C2']

切片取得的單元格範圍如下:

以行或列為單位也可以:

# 取出 C 列
>>> colC = ws['C']

# 取出 C 至 D 列
>>> col_range = ws['C:D']

# 取出第 10 行
>>> row10 = ws[10]

# 取出第 5 至 10 行
>>> row_range = ws[5:10]

使用 iter_rows 方法也可以:

# 從第 1 行開始遍歷,直到第 2 行,每行最多返回 3 列
>>> for row in ws.iter_rows(min_row=1, max_row=2, max_col=3):
...    for cell in row:
...        print(cell)
<Cell Sheet1.A1>
<Cell Sheet1.B1>
<Cell Sheet1.C1>
<Cell Sheet1.A2>
<Cell Sheet1.B2>
<Cell Sheet1.C2>

如需遍歷表格所有行或列,可以使用相關屬性。使用 rows 屬性遍歷所有行:

>>> ws = wb.active
>>> ws['C9'] = 'hello world'
>>> tuple(ws.rows)
((<Cell Sheet.A1>, <Cell Sheet.B1>, <Cell Sheet.C1>),
(<Cell Sheet.A2>, <Cell Sheet.B2>, <Cell Sheet.C2>),
(<Cell Sheet.A3>, <Cell Sheet.B3>, <Cell Sheet.C3>),
(<Cell Sheet.A4>, <Cell Sheet.B4>, <Cell Sheet.C4>),
(<Cell Sheet.A5>, <Cell Sheet.B5>, <Cell Sheet.C5>),
(<Cell Sheet.A6>, <Cell Sheet.B6>, <Cell Sheet.C6>),
(<Cell Sheet.A7>, <Cell Sheet.B7>, <Cell Sheet.C7>),
(<Cell Sheet.A8>, <Cell Sheet.B8>, <Cell Sheet.C8>),
(<Cell Sheet.A9>, <Cell Sheet.B9>, <Cell Sheet.C9>))

使用 columns 屬性遍歷所有列:

>>> tuple(ws.columns)
((<Cell Sheet.A1>,
<Cell Sheet.A2>,
<Cell Sheet.A3>,
<Cell Sheet.A4>,
<Cell Sheet.A5>,
<Cell Sheet.A6>,
...
<Cell Sheet.B7>,
<Cell Sheet.B8>,
<Cell Sheet.B9>),
(<Cell Sheet.C1>,
<Cell Sheet.C2>,
<Cell Sheet.C3>,
<Cell Sheet.C4>,
<Cell Sheet.C5>,
<Cell Sheet.C6>,
<Cell Sheet.C7>,
<Cell Sheet.C8>,
<Cell Sheet.C9>))

資料儲存

Excel 表格通過單元格儲存資料,直接賦值即可:

>>> cell.value = 'hello, world'
>>> print(cell.value)
'hello, world'

>>> cell2.value = 3.14
>>> print(cell2.value)
3.14

與此同時,還可以給單元格附加型別以及格式化資訊,建立工作簿時需要指定 guess_types 引數:

>>> wb = Workbook(guess_types=True)

這樣一來,文字(包括百分比)將自動轉換成浮點數:

>>> cell.value = '31.50'
>>> print(cell.value)
31.5

>>> cell2.value = '12%'
>>> print(cell2.value)
0.12

日期可以直接由原生的 datetime 物件來設定:

>>> import datetime
>>> cell.value = datetime.datetime.now()
>>> print cell.value
datetime.datetime(2010, 9, 10, 22, 25, 18)

儲存至檔案

最保險的儲存方式是呼叫 save 方法儲存到指定檔案:

>>> wb = Workbook()
>>> wb.save('balances.xlsx')

警告

這個操作將覆蓋已存在的檔案,沒有任何提示!

藉助 template 屬性,可以將工作表儲存成模板文件:

>>> wb = load_workbook('document.xlsx')
>>> wb.template = True
>>> wb.save('document_template.xltx')

或者儲存成普通文件:

>>> wb = load_workbook('document_template.xltx')
>>> wb.template = False
>>> wb.save('document.xlsx', as_template=False)

儲存至流

FlaskDjangoWeb 應用,可能需要將檔案儲存到流( stream )。 藉助一個臨時檔案( NamedTemporaryFile )可以輕鬆實現:

>>> from tempfile import NamedTemporaryFile
>>> from openpyxl import Workbook
>>> wb = Workbook()

# 先儲存到臨時檔案,再將檔案內容讀出
>>> with NamedTemporaryFile() as tmp:
...     wb.save(tmp.name)
...     tmp.seek(0)
...     stream = tmp.read()

附錄

更多 Python 技術文章請訪問:Python語言小