python openpyxl 操作 excel


初識與安裝

Openpyxl is a Python library for reading and writing Excel 2010 xlsx/xlsm/xltx/xltm files.

安裝

$ pip install openpyxl

  

一個簡單創建例子

from openpyxl import Workbook 
wb = Workbook()
# 激活 worksheet
ws = wb.active
# 數據可以直接分配到單元格中
ws['A1'] = 42
# 可以附加行,從第一列開始附加
ws.append([1, 2, 3])
# Python 類型會被自動轉換
import datetime
ws['A3'] = datetime.datetime.now().strftime("%Y-%m-%d")
# 保存文件
wb.save("sample.xlsx")

打開查看Excel如下:

  

workbook

There is no need to create a file on the filesystem to get started with openpyxl. Just import the Workbook class and start using it.

from openpyxl import Workbook
wb = Workbook()

  

worksheet

A workbook至少創建一個worksheet.

通過openpyxl.workbook.Workbook.active()得到worksheet.
ws = wb.active
注意:
該方法使用_active_sheet_index屬性, 默認會設置0,也就是第一個worksheet。除非手動修改,否則使用active方法得到都是第一個worksheet。
你也可以創建worksheets,通過 openpyxl.workbook.Workbook.create_sheet() 方法:

>>> ws1 = wb.create_sheet("Mysheet") #插入到最后(default)
#或者
>>> ws2 = wb.create_sheet("Mysheet", 0) #插入到最開始的位置

創建的sheet的名稱會自動創建,按照sheet,sheet1,sheet2自動增長,通過title屬性可以修改其名稱。
ws.title = "New Title"
默認的sheet的tab是白色的,可以通過 RRGGBB顏色來修改sheet_properties.tabColor屬性從而修改sheet tab按鈕的顏色:
ws.sheet_properties.tabColor = "1072BA"
當你設置了sheet的名稱,可以將其看成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

查看workbook中的所有worksheets名稱:openpyxl.workbook.Workbook.get_sheet_names()

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

遍歷worksheets:

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

  

操作數據

訪問單元格

單元格可以看作是worksheet的key,通過key去訪問單元格中的數據

>>> c = ws['A4']

直接返回A4單元格,如果不存在則會自動創建一個。

指定單元格的值

>>> ws['A4'] = 4 #直接賦值

使用openpyxl.worksheet.Worksheet.cell()方法操作某行某列的某個值:

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

注意:

  1. 當worksheet在內存中被創建時,是沒有包含cells的,cells是在首次訪問時創建.
  2. 可以循環在內存中創建cells,這時不指定他們的值也會創建該cells些:(創建100x100cells)
>>> for i in range(1,101):
...        for j in range(1,101):
...            ws.cell(row=i, column=j)

訪問許多cells

通過切片Ranges指定許多cells

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

同樣也可以Ranges rows 或者columns :

>>> colC = ws['C']
>>> col_range = ws['C:D']
>>> row10 = ws[10]
>>> row_range = ws[5:10]

也可以使用 openpyxl.worksheet.Worksheet.iter_rows() 方法:(需要指定行->行,截止列)

>>> for row in ws.iter_rows(min_row=1, max_col=3, max_row=2):
...    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.iter_cols() 方法:(需要指定列->列,截止行)
>>> for row in ws.iter_rows(min_row=1, max_col=3, max_row=2):
...    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'
>>> 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>))

或者 openpyxl.worksheet.Worksheet.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>))

Saving to a file

最簡單最安全的方法保存workbook是使用openpyxl.workbook.Workbook對象的 openpyxl.workbook.Workbook.save()方法:

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

保存的默認位置在python的根目錄下。
注意:會自動覆蓋已經存在文件名的文件。

Loading from a file

像寫一樣我們可以導入openpyxl.load_workbook()已經存在的workbook:

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

  

常用實例

詳情參考官方使用文檔:

http://openpyxl.readthedocs.io/en/default/usage.html

寫入例子一

#!/usr/bin/env python 
# -*- coding: utf-8 -*-

from openpyxl import Workbook

wb = Workbook()
# 激活 worksheet
ws = wb.active
# 數據可以直接分配到單元格中
ws['A1'] = 42
# 可以附加行,從第一列開始附加
ws.append([1, 2, 3])
# Python 類型會被自動轉換
import datetime

ws['A3'] = datetime.datetime.now().strftime("%Y-%m-%d")
# 保存文件
wb.save("sample.xlsx")

寫入例子二

#!/usr/bin/env python 
# -*- coding: utf-8 -*-
"""
http://openpyxl.readthedocs.io/en/default/usage.html
"""

# workbook相關
from openpyxl import Workbook
from openpyxl.compat import range
from openpyxl.utils import get_column_letter

wb = Workbook()

dest_filename = 'empty_book.xlsx'

ws1 = wb.active
ws1.title = "range names"

for row in range(1, 40):
    ws1.append(range(600))

ws2 = wb.create_sheet(title="Pi")

ws2['F5'] = 3.14

ws3 = wb.create_sheet(title="Data")
for row in range(10, 20):
    for col in range(27, 54):
        _ = ws3.cell(column=col, row=row, value="{0}".format(get_column_letter(col)))
print(ws3['AA10'].value)
wb.save(filename=dest_filename)

讀取例子一

#!/usr/bin/env python 
# -*- coding: utf-8 -*-

from openpyxl.reader.excel import load_workbook
import json

# 讀取excel2007文件
wb = load_workbook(filename=r'test_book.xlsx')

# 顯示有多少張表
print "Worksheet range(s):", wb.get_named_ranges()
print "Worksheet name(s):", wb.get_sheet_names()

# 取第一張表
sheetnames = wb.get_sheet_names()
ws = wb.get_sheet_by_name(sheetnames[0])

# 顯示表名,表行數,表列數
print "Work Sheet Titile:", ws.title
print "Work Sheet Rows:", ws.max_row
print "Work Sheet Cols:", ws.max_column


# 建立存儲數據的字典
data_dic = {}

# 把數據存到字典中
for rx in range(1, ws.max_row + 1):
temp_list = []
pid = rx
w1 = ws.cell(row=rx, column=1).value
w2 = ws.cell(row=rx, column=2).value
w3 = ws.cell(row=rx, column=3).value
w4 = ws.cell(row=rx, column=4).value
temp_list = [w1, w2, w3, w4]

data_dic[pid] = temp_list

# 打印字典數據個數
print 'Total:%d' % len(data_dic)
print json.dumps(data_dic, encoding="UTF-8", ensure_ascii=False)

 讀取結果:

Worksheet range(s): []
Worksheet name(s): [u'\u6d3b\u52a8\u8868', u'\u7528\u6237\u4fe1\u606f', u'Sheet3']
Work Sheet Titile: 活動表
Work Sheet Rows: 3
Work Sheet Cols: 5
Total:3
{"1": ["張三", 18, "男", "廣州"], "2": ["李四", 20, "女", "湖北"], "3": ["王五", 25, "女", "北京"]}

  

使用公式

>>> from openpyxl import Workbook
>>> wb = Workbook()
>>> ws = wb.active
>>> # add a simple formula
>>> ws["A1"] = "=SUM(1, 1)"
>>> wb.save("formula.xlsx")

 

 

 


 

***微信掃一掃,關注“python測試開發圈”,了解更多測試教程!***


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM