最近在網(wǎng)上爬取奧運(yùn)項(xiàng)目資料矛市,并寫入Excel中界牡。在寫到Excel中是用到了OpenPyXL淤齐,翻譯了一部分自己用到的知識(shí)點(diǎn)。
創(chuàng)建一個(gè)工作簿
- 使用openpyxl沒有必要先在系統(tǒng)中新建一個(gè).xlsx潮饱,我們需要做的只需要引入Workbook這個(gè)類,接著開始調(diào)用它诫给。
>>> from openpyxl import Workbook
>>> wb = Workbook()
- 一個(gè)工作簿(workbook)在創(chuàng)建的時(shí)候同時(shí)至少也新建了一張工作表(worksheet)香拉。你可以通過openpyxl.workbook.Workbook.active()調(diào)用得到正在運(yùn)行的工作表。
>>> ws = wb.active
注意:該函數(shù)調(diào)用工作表的索引(_active_sheet_index)中狂,默認(rèn)是0凫碌。除非你修改了這個(gè)值,否則你使用該函數(shù)一直是在對(duì)第一張工作表進(jìn)行操作胃榕。
- 使用openpyxl.workbook.Workbook.create_sheet()新建一張表
>>> ws1 = wb.create_sheet() #默認(rèn)插在工作簿末尾
# or
>>> ws2 = wb.create_sheet(0) # 插入在工作簿的第一個(gè)位置
- 在創(chuàng)建工作表的時(shí)候系統(tǒng)自動(dòng)命名盛险。他們按照序列依次命名 (Sheet, Sheet1, Sheet2, ...)。你可以通過調(diào)用下面的屬性修改工作表的名稱:
ws.title = "New Title"
- 標(biāo)簽欄的背景色默認(rèn)為白色勋又。你可以通過提供一個(gè)RRGGBB顏色碼改變標(biāo)簽欄的字體顏色:
ws.sheet_properties.tabColor = "1072BA"
- 一旦你獲取工作表的名字苦掘,你可以通過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']
你也可以循環(huán)得到所有的工作表
>>> for sheet in wb:
print(sheet.title)
操作數(shù)據(jù)
使用一個(gè)單元格
現(xiàn)在我們已經(jīng)知道如何使用一張工作表了楔壤,下面我我們開始修改工作表中單元格的內(nèi)容
- 單元格可以直接根據(jù)他們的索引直接獲得
>>> c = ws['A4']
通過上述的語句鹤啡,將返回在A4處的單元格,如果不存在將在A4新建一個(gè)挺邀。* * 單元格的值也可以直接賦值
>>> ws['A4'] = 4
- 還提供** openpyxl.worksheet.Worksheet.cell()
**方法獲取單元格
>>> c = ws.cell('A4')
- 也可以根據(jù)行列值獲取單元格
>>> d = ws.cell(row = 4, column = 2)
注意:當(dāng)一個(gè)工作表被創(chuàng)建是揉忘,其中不包含單元格。只有當(dāng)單元格被獲取是才被創(chuàng)建端铛。這種方式我們不會(huì)創(chuàng)建我們從不會(huì)使用的單元格泣矛,從而減少了內(nèi)存消耗。
警告:由于上述特性禾蚕,你如果遍歷了單元格而非想要使用它們也將會(huì)在內(nèi)存當(dāng)中創(chuàng)建您朽。比如下面:
>>> for i in range(1,101):
for j in range(1,101):
ws.cell(row = i, column = j)
上述代碼將會(huì)在內(nèi)存中創(chuàng)建100*100個(gè)單元格。
當(dāng)然,這里也有方法來清理這些不想要的單元格哗总,在后續(xù)我們將會(huì)介紹几颜。
使用多個(gè)單元格
- 使用切片獲取多個(gè)單元格
>>> cell_range = ws['A1':'C2']
- 使用**openpyxl.worksheet.Worksheet.iter_rows()
**方法獲得多個(gè)單元格
>>> 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>))
數(shù)據(jù)存儲(chǔ)
- 一旦我們有一個(gè)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
保存到文件
- 保存工作簿最簡(jiǎn)單最安全的方式是使用openpyxl.workbook.Workbook的**openpyxl.workbook.Workbook.save()
**方法
>>> wb = Workbook()
>>> wb.save('balances.xlsx')
蛋哭!特別警告:這個(gè)操作將會(huì)在沒有認(rèn)識(shí)提示的情況下用現(xiàn)在寫的內(nèi)容,覆蓋掉原文件中的所有內(nèi)容
- 你也可以 as_template=True涮母,將文件保存稱為一個(gè)模板
>>> wb = load_workbook('document.xlsx')
>>> wb.save('document_template.xltx', as_template=True)
- 如果as_template=False(默認(rèn))谆趾,則將文件或模板保存為文件
>>> 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)
警告:在保存文件到文件模板中的時(shí)候你應(yīng)該監(jiān)控?cái)?shù)據(jù)的屬性和文件擴(kuò)展名,反之亦然叛本;否則沪蓬,你得到的工作簿可能無法打開。
比如下面的:
>>> 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
從文件中導(dǎo)入
- 和寫入文件的方式相同来候,你可以引入openpyxl.load_workbook()來打開一個(gè)已經(jīng)存在的工作簿
>>> from openpyxl import load_workbook
>>> wb2 = load_workbook('test.xlsx')
>>> print wb2.get_sheet_names()
['Sheet2', 'New Title', 'Sheet1']