最近在網上爬取奧運項目資料,并寫入Excel中。在寫到Excel中是用到了OpenPyXL,翻譯了一部分自己用到的知識點。
創建一個工作簿
- 使用openpyxl沒有必要先在系統中新建一個.xlsx,我們需要做的只需要引入Workbook這個類,接著開始調用它。
>>> from openpyxl import Workbook
>>> wb = Workbook()
- 一個工作簿(workbook)在創建的時候同時至少也新建了一張工作表(worksheet)。你可以通過openpyxl.workbook.Workbook.active()調用得到正在運行的工作表。
>>> ws = wb.active
注意:該函數調用工作表的索引(_active_sheet_index),默認是0。除非你修改了這個值,否則你使用該函數一直是在對第一張工作表進行操作。
- 使用openpyxl.workbook.Workbook.create_sheet()新建一張表
>>> ws1 = wb.create_sheet() #默認插在工作簿末尾
# or
>>> ws2 = wb.create_sheet(0) # 插入在工作簿的第一個位置
- 在創建工作表的時候系統自動命名。他們按照序列依次命名 (Sheet, Sheet1, Sheet2, ...)。你可以通過調用下面的屬性修改工作表的名稱:
ws.title = "New Title"
- 標簽欄的背景色默認為白色。你可以通過提供一個RRGGBB顏色碼改變標簽欄的字體顏色:
ws.sheet_properties.tabColor = "1072BA"
修改顏色效果.png
- 一旦你獲取工作表的名字,你可以通過workbook的key或者** openpyxl.workbook.Workbook.get_sheet_by_name()
**方法得到該工作表
>>> ws3 = wb["New Title"]
>>> ws4 = wb.get_sheet_by_name("New Title")
>>> ws is ws3 is ws4
True
- 你可以通過**openpyxl.workbook.Workbook.get_sheet_names()
**方法得到工作簿的所有工作表。
>>> print(wb.get_sheet_names())
['Sheet2', 'New Title', 'Sheet1']
你也可以循環得到所有的工作表
>>> for sheet in wb:
print(sheet.title)
操作數據
使用一個單元格
現在我們已經知道如何使用一張工作表了,下面我我們開始修改工作表中單元格的內容
- 單元格可以直接根據他們的索引直接獲得
>>> c = ws['A4']
通過上述的語句,將返回在A4處的單元格,如果不存在將在A4新建一個。* * 單元格的值也可以直接賦值
>>> ws['A4'] = 4
- 還提供** openpyxl.worksheet.Worksheet.cell()
**方法獲取單元格
>>> c = ws.cell('A4')
- 也可以根據行列值獲取單元格
>>> d = ws.cell(row = 4, column = 2)
注意:當一個工作表被創建是,其中不包含單元格。只有當單元格被獲取是才被創建。這種方式我們不會創建我們從不會使用的單元格,從而減少了內存消耗。
警告:由于上述特性,你如果遍歷了單元格而非想要使用它們也將會在內存當中創建。比如下面:
>>> for i in range(1,101):
for j in range(1,101):
ws.cell(row = i, column = j)
上述代碼將會在內存中創建100*100個單元格。
當然,這里也有方法來清理這些不想要的單元格,在后續我們將會介紹。
使用多個單元格
- 使用切片獲取多個單元格
>>> cell_range = ws['A1':'C2']
- 使用**openpyxl.worksheet.Worksheet.iter_rows()
**方法獲得多個單元格
>>> tuple(ws.iter_rows('A1:C2'))
((<Cell Sheet1.A1>, <Cell Sheet1.B1>, <Cell Sheet1.C1>),
(<Cell Sheet1.A2>, <Cell Sheet1.B2>, <Cell Sheet1.C2>))
>>> for row in ws.iter_rows('A1:C2'):
for cell in row:
print cell
<Cell Sheet1.A1>
<Cell Sheet1.B1>
<Cell Sheet1.C1>
<Cell Sheet1.A2>
<Cell Sheet1.B2>
<Cell Sheet1.C2>
- 如果你需要迭代文件中所有的行或者列,你可以使用openpyxl.worksheet.Worksheet.rows()
>>> ws = wb.active
>>> ws['C9'] = 'hello world'
>>> 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>))
或者使用openpyxl.worksheet.Worksheet.columns()方法
>>> 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>))
數據存儲
- 一旦我們有一個openpyxl.cell.Cell,我們可以直接為該單元格賦值
>>> c.value = 'hello, world'
>>> print(c.value)
'hello, world'
>>> d.value = 3.14
>>> print(d.value)
3.14
- 你也可以使用Python中的其他類型和格式
>>> wb = Workbook(guess_types=True)
>>> c.value = '12%'
>>> print(c.value)
0.12
>>> import datetime
>>> d.value = datetime.datetime.now()
>>> print d.value
datetime.datetime(2010, 9, 10, 22, 25, 18)
>>> c.value = '31.50'
>>> print(c.value)
31.5
保存到文件
- 保存工作簿最簡單最安全的方式是使用openpyxl.workbook.Workbook的**openpyxl.workbook.Workbook.save()
**方法
>>> wb = Workbook()
>>> wb.save('balances.xlsx')
!特別警告:這個操作將會在沒有認識提示的情況下用現在寫的內容,覆蓋掉原文件中的所有內容
- 你也可以 as_template=True,將文件保存稱為一個模板
>>> wb = load_workbook('document.xlsx')
>>> wb.save('document_template.xltx', as_template=True)
- 如果as_template=False(默認),則將文件或模板保存為文件
>>> wb = load_workbook('document_template.xltx')
>>> wb.save('document.xlsx', as_template=False)
>>> wb = load_workbook('document.xlsx')
>>> wb.save('new_document.xlsx', as_template=False)
警告:在保存文件到文件模板中的時候你應該監控數據的屬性和文件擴展名,反之亦然;否則,你得到的工作簿可能無法打開。
比如下面的:
>>> wb = load_workbook('document.xlsx')
>>> # Need to save with the extension *.xlsx
>>> wb.save('new_document.xlsm')
>>> # MS Excel can't open the document
>>>
>>> # or
>>>
>>> # Need specify attribute keep_vba=True
>>> wb = load_workbook('document.xlsm')
>>> wb.save('new_document.xlsm')
>>> # MS Excel can't open the document
>>>
>>> # or
>>>
>>> wb = load_workbook('document.xltm', keep_vba=True)
>>> # If us need template document, then we need specify extension as *.xltm.
>>> # If us need document, then we need specify attribute as_template=False.
>>> wb.save('new_document.xlsm', as_template=True)
>>> # MS Excel can't open the document
從文件中導入
- 和寫入文件的方式相同,你可以引入openpyxl.load_workbook()來打開一個已經存在的工作簿
>>> from openpyxl import load_workbook
>>> wb2 = load_workbook('test.xlsx')
>>> print wb2.get_sheet_names()
['Sheet2', 'New Title', 'Sheet1']