最全總結 | 聊聊 Python 辦公自動化之 Excel(中)
阿新 • • 發佈:2020-11-11
![image](https://upload-images.jianshu.io/upload_images/1466987-00af61a74363a216?imageMogr2/auto-orient/strip%7CimageView2/2/w/1240)
## 1\. 前言
上一篇文章中,我們聊到使用 xlrd、xlwt、xlutils 這一組合操作 Excel 的方法
[最全總結 | 聊聊 Python 辦公自動化之 Excel(上)](http://mp.weixin.qq.com/s?__biz=MzU1OTI0NjI1NQ==&mid=2247486752&idx=1&sn=bc58a3127bad72f2210817c3b4087801&chksm=fc1b73e0cb6cfaf6b1dbe8943d13a805d384ae74f043720a1d048c498405cc1ada93437f58ce&scene=21#wechat_redirect)
本篇文章將繼續聊另外一種方式,即:openpyxl
不得不說,openpyxl 更強大!
它支援 xlsx 格式的表格檔案,並且支援 Numpy、Pandas 等包,可用於繪製圖表
## 2\. 準備
首先,我們需要安裝依賴包
```
# 安裝依賴包
pip3 install openpyxl
```
## 3\. 讀取資料
使用 openpyxl 中的 load_workbook(filepath) 載入本地一個 Excel 檔案,返回結果是一個工作簿物件
```
import openpyxl
# 載入本地的Excel檔案
wb = openpyxl.load_workbook(file_path)
```
利用工作簿物件,可以獲取所有的 Sheet 名稱及 Sheet 列表
```
def get_all_sheet_names(wb):
"""
獲取所有sheet的名稱
:param wb:
:return:
"""
# sheet名稱列表
sheet_names = wb.sheetnames
return sheet_names
def get_all_sheet(wb):
"""
獲取所有的sheet
:param wb:
:return:
"""
# sheet名稱列表
sheet_names = get_all_sheet_names(wb)
# 所有sheet
sheets = []
for sheet_name in sheet_names:
sheet = wb[sheet_name]
sheets.append(sheet)
return sheets
```
工作簿物件提供了 active 屬性,用於快速獲取當前選擇的 Sheet
```
def get_current_sheet(wb):
"""
獲取當前選擇的sheet,預設是最後一個sheet
:param wb:
:return:
"""
# 當前選中的sheet
current_sheet = wb.active
return current_sheet
```
另外,也可以通過 Sheet 名稱去獲取某一個特定的 Sheet 物件
```
def get_sheet_by_name(wb, sheet_name):
"""
通過sheetname去查詢某一個sheet
:param wb:
:param sheet_name:
:return:
"""
sheet_names = get_all_sheet_names(wb)
if sheet_name in sheet_names:
result = wb[sheet_name]
else:
result = None
return result
```
使用 sheet.max_row 和 sheet.max_column 可以獲取當前 Sheet 中的資料行數和列數
```
def get_row_and_column_num(sheet):
"""
獲取sheet的行數和列數
:param sheet:
:return:
"""
# 行數
row_count = sheet.max_row
# 列數
column_count = sheet.max_column
return row_count, column_count
# 行數和列數
row_count, column_count = get_row_and_column_num(sheet)
print('行數和列數分別為:', row_count, column_count)
```
openpyxl 提供 2 種方式來定位一個單元格,分別是:
* 數字索引,從 1 開始
數字索引:行數字索引、列數字索引
比如:row_index=1,column_index=1
* 行和列組成的字串索引
字串索引:列由字母組成 + 行索引
比如:A1 對應第一行、第一列的單元格
並且,openpyxl.utils 提供了方法,便於 列索引 在兩者之間進行轉換
```
from openpyxl.utils import get_column_letter, column_index_from_string
def column_num_to_str(num):
"""
Excel索引列從數字轉為字母
:param num:
:return:
"""
return get_column_letter(num)
def column_str_to_num(str):
"""
Excel索引列,從字母轉為數字
:param str:
:return:
"""
return column_index_from_string(str)
```
單元格的獲取,同樣可以通過上面 2 種索引方式來獲取
```
def get_cell(sheet, row_index, column_index):
"""
獲取單元格
:param sheet:
:param row_index:
:param column_index:
:return:
"""
# openpyxl索引都是從1開始計數,這與xlrd有所不同
# 獲取某一個單元格(二選一)
# 比如:獲取A1單元格的資料,即第一個行、第一列的資料
# cell_one = sheet['A1']
cell_one = sheet.cell(row=row_index, column=column_index)
return cell_one
```
在日常處理 Excel 資料過程中,可能需要判斷單元格資料型別,而 openpyxl 並沒有提供現成的方法
這裡,我們可以通過單元格物件的 value 屬性拿到值,接著使用 isinstance 方法判斷資料型別
```
def get_cell_value_and_type(cell):
"""
獲取某一個cell的內容及資料型別
:param cell:
:return:
"""
# 單元格的值
cell_value = cell.value
# 單元格的型別
cell_type = get_cell_value_type(cell_value)
return cell_value, cell_type
def get_cell_value_type(cell_value):
"""
獲取資料型別
:param cell_value:
:return:
"""
# 其中
# 0:空
# 1:數字
# 2:字串
# 3:日期
# 4:其他
if not cell_value:
cell_type = 0
elif isinstance(cell_value, int) or isinstance(cell_value, float):
cell_type = 1
elif isinstance(cell_value, str):
cell_type = 2
elif isinstance(cell_value, datetime.datetime):
cell_type = 3
else:
cell_type = 4
return cell_type
```
單獨獲取某一行[列]的資料,可以使用下面的方式:
```
def get_row_cells_by_index(sheet, row_index):
"""
通過行索引,獲取某一行的單元格
:param row_index:
:return:
"""
# 注意:第一列從1開始
row_cells = sheet[row_index]
return row_cells
def get_column_cells_by_index(sheet, column_index):
"""
通過列索引,獲取某一列的單元格
"""
# 數字轉為字母
column_index_str = column_num_to_str(column_index)
# 獲取某一列的資料
column_cells = sheet[column_index_str]
return column_cells
```
需要注意的是,獲取某一行的資料需要傳入數字索引;而對於列資料的獲取,必須傳入字串索引
和 Python 列表範圍取值類似,openpyxl 同樣支援使用 : 符號拿到某個範圍內的資料行[列]
```
def get_rows_by_range(sheet, row_index_start, row_index_end):
"""
通過範圍去選擇行範圍
比如:選擇第2行到第4行的所有資料,返回值為元組
:param sheet:
:param row_index_start:
:param row_index_end:
:return:
"""
rows_range = sheet[row_index_start:row_index_end]
return rows_range
def get_columns_by_range(sheet, column_index_start, column_index_end):
"""
通過範圍去選擇列範圍
比如:選擇第2列到第4列的所有資料,返回值為元組
:param sheet:
:param column_index_start:
:param column_index_end:
:return:
"""
columns_range = sheet[column_num_to_str(column_index_start):column_num_to_str(column_index_end)]
return columns_range
```
## 4\. 寫入資料
要寫入資料到 Excel 表格
首先,使用 openpyxl.Workbook() 建立一個 Excel 工作簿物件
接著,使用工作簿物件的 create_sheet() 新建一個 Sheet
```
# 建立一個Excel工作簿
# 注意:每次新建一個Excel檔案,都會預設生成一個名稱為【Sheet】的工作表Sheet
wb = openpyxl.Workbook()
# 建立一個新的sheet,預設被插到尾部
# new_sheet = wb.create_sheet('新的Sheet')
# 也可以通過第二個引數:index來指定插入的位置
# 比如:插入到開頭
new_sheet = wb.create_sheet('新的Sheet', 0)
```
預設建立的 Sheet 被插入到最後一個位置,第 2 個引數可以指定 Sheet 插入的位置
Sheet 標籤的背景色同樣支援修改,使用 sheet_properties.tabColor 指定 RGB 顏色值
比如,要設定某一個 Sheet 的背景色為紅色,只需要先查詢到對應的 Sheet,然後指定顏色值為 FF0000 即可
```
def set_sheet_bg_color(sheet, rgb_value):
"""
設定Sheet標籤的顏色
:param rgb_value:
:return:
"""
# 設定Sheet底部按鈕的顏色(RRGGBB)
sheet.sheet_properties.tabColor = rgb_value
# 設定Sheet的背景色(紅色)
set_sheet_bg_color(new_sheet, 'FF0000')
```
openpyxl 支援行列數字索引、字串索引以這 2 種方式寫入資料到單元格中
```
def write_value_to_cell_with_num(sheet, row_index, column_index, value):
"""
按行索引、列索引寫入資料
:param shell:
:param row_index: 行索引
:param column_index: 列索引
:param value:
:return:
"""
# 二選一
sheet.cell(row=row_index, column=column_index, value=value)
# shell.cell(row=row_index, column=column_index).value = value
def write_value_to_cell_with_index_str(sheet, index_str, value):
"""
按字母位置,寫入資料到對應單元格
:param shell:
:param index_str: 字母對應的單元格位置
:param value:
:return:
"""
sheet[index_str] = value
```
在單元格中插入圖片也很簡單,openpyxl 提供的 add_image() 方法
引數有 2 個,分別是:圖片物件、單元格字串索引
為了便於使用,我們可以將列索引進行轉換,然後封裝成兩個插入圖片的方法
```
from openpyxl.drawing.image import Image
def insert_img_to_cell_with_num(sheet, image_path, row_index, column_index):
"""
往單元格中插入圖片
:param sheet:
:param image_path:
:param row_index:
:param column_index:
:return:
"""
# 通過行索引、列索引,獲取到字母索引
index_str = column_num_to_str(column_index) + str(row_index)
insert_img_to_cell_with_str(sheet, image_path, index_str)
def insert_img_to_cell_with_str(sheet, image_path, index_str):
"""
往單元格中插入圖片
:param sheet:
:param image_path:
:param index_str:
:return:
"""
sheet.add_image((image_path), index_str)
```
最後,呼叫工作簿物件的 save() 方法,將資料真實寫入到 Excel 檔案中
```
# 注意:必須要寫入,才能真實的儲存到檔案中
wb.template = False
wb.save('new.xlsx')
```
## 5\. 修改資料
修改資料包含:單元格資料的修改、單元格樣式的修改
對於單元格資料的修改,只需要先讀取工作簿物件,查詢到要操作的 Sheet 物件,然後呼叫上面的方法修改單元格資料,最後呼叫 save() 函式儲存覆蓋即可
```
def modify_excel(self, file_path):
"""
修改本地Excel檔案中資料
:param file_path:
:return:
"""
# 讀取本地Excel檔案
wb = openpyxl.load_workbook(file_path)
# 讀取某一個sheet
sheet = wb['第一個Sheet']
print(sheet)
# 直接修改某一個單元格的資料
write_value_to_cell_with_num(sheet, 1, 1, '姓名1')
# 儲存並覆蓋
wb.save(file_path)
```
單元格樣式包含:字型樣式、單元格背景樣式、邊框樣式、對齊方式等
以常見的字型樣式、對齊方式為例
首先,使用 openpyxl 中的 Font 類建立一個物件,指定字型名稱、字型大小、是否加粗、是否斜體、顏色、下劃線等
```
from openpyxl.styles import Font
# 字型格式
# 指定字型型別、大小、是否加粗、顏色等
font0 = Font(name='Calibri',
size=20,
bold=False,
italic=False,
vertAlign=None,
underline='none',
strike=False,
color='FF00FF00')
```
接著,構建一個 Alignment 物件,指定單元格的對齊方式
```
from openpyxl.styles import Font,Alignment
# 單元格對齊方式
alignment0 = Alignment(horizontal='center',
vertical='bottom',
text_rotation=0,
wrap_text=False,
shrink_to_fit=False,
indent=0)
```
最後,使用單元格物件的 font/alignment 屬性,將字型樣式和對齊方式設定進去即可
```
# 設定屬性樣式(字型、對齊方式)
sheet['A1'].font = font0
sheet['A1'].alignment = alignment0
```
## 6\. 進階用法
接下來,聊聊幾個常用的進階用法
1、獲取可見及隱藏的 Sheet
通過判斷 Sheet 物件的 sheet_state 屬性值,可以判斷當前 Sheet 是顯示還是隱藏
當值為 visible 時,代表 Sheet 是顯示的
當值是 hidden 時,代表這個 Sheet 被隱藏了
```
def get_all_visiable_sheets(wb):
"""
獲取工作簿中所有可見的sheet
:param wb:
:return:
"""
return [sheet for sheet in get_all_sheet(wb) if sheet.sheet_state == 'visible']
def get_all_hidden_sheets(wb):
"""
獲取工作簿中所有隱藏的sheet
:param wb:
:return:
"""
return [sheet for sheet in get_all_sheet(wb) if sheet.sheet_state == 'hidden']
```
2、獲取隱藏/顯示的行索引列表、列索引列表
受限於篇幅,這裡以獲取所有顯示/隱藏的行索引列表為例
遍歷 Sheet 物件的 row_dimensions 屬性值,通過判斷行屬性的 hidden 值,判斷當前行是否隱藏或顯示
```
def get_all_rows_index(sheet, hidden_or_visiable):
"""
獲取所有隱藏/顯示的行
:param hidden_or_visiable: True:隱藏;False:顯示
:param sheet:
:return:
"""
# 遍歷行
# 隱藏的索引
hidden_indexs = []
# 所有隱藏的行索引
for row_index, rowDimension in sheet.row_dimensions.items():
if rowDimension.hidden:
hidden_indexs.append(row_index)
# 所有顯示的行索引
visiable_indexs = [index + 1 for index in range(get_row_and_column_num(sheet)[0]) if index + 1 not in hidden_indexs]
# 隱藏或者顯示的行索引列表
return hidden_indexs if hidden_or_visiable else visiable_indexs
```
3、獲取單元格字型顏色及單元格背景顏色
單元格物件的 font.color.rgb、fill.fgColor.rgb 屬性值分別代表字型顏色值、單元格背景顏色
```
def get_cell_font_color(sheet, row_index, column_index):
"""
獲取單元格字型的顏色
:param sheet:
:param row_index:行索引
:param column_index:列索引
:return:
"""
cell_color = sheet.cell(row_index, column_index).font.color
if cell_color:
return sheet.cell(row_index, column_index).font.color.rgb
else:
# 顏色不存在,可能單元格沒有資料
return None
def get_cell_bg_color(sheet, row_index, column_index):
"""
獲取單元格背景的顏色
:param sheet:
:param row_index:行索引
:param column_index:列索引
:return:
"""
return sheet.cell(row_index, column_index).fill.fgColor.rgb
```
## 7\. 最後
可以發現,openpyxl 相比 xlrd/xlwt,提供了大量實用的 API,功能更強大,並且完美支援 xlsx!
受限於篇幅,文中只展示了部分功能和程式碼,更加複雜的功能,比如:單元格合併、單元格完整樣式操作,我已經封裝成方法上傳到後臺
要獲取全部原始碼,關注公眾號「 **AirPython** 」,後臺回覆「 **excel** 」即可獲得全部原始碼
如果你覺得文章還不錯,請大家 **點贊、分享、留言**下,因為這將是我持續輸出更多優質文章的最強動力!
**推薦閱讀**
[最全總結 | 聊聊 Python 辦公自動化之 Excel(上)](http://mp.weixin.qq.com/s?__biz=MzU1OTI0NjI1NQ==&mid=2247486752&idx=1&sn=bc58a3127bad72f2210817c3b4087801&chksm=fc1b73e0cb6cfaf6b1dbe8943d13a805d384ae74f043720a1d048c498405cc1ada93437f58ce&scene=21#wechat_redirect)
[最全總結 | 聊聊 Python 資料處理全家桶(配置篇)](http://mp.weixin.qq.com/s?__biz=MzU1OTI0NjI1NQ==&mid=2247486592&idx=1&sn=78718ab892a8e22b8ac334ed773f450d&chksm=fc1b7240cb6cfb56884827ab62f2c403fb733a24dd25d1dae2d16dd6c18e259389a838c94f15&scene=21#wechat_redirect)
[我用幾行 Python 自動化指令碼完美解決掉了小姐姐的微信焦慮感](http://mp.weixin.qq.com/s?__biz=MzU1OTI0NjI1NQ==&mid=2247486280&idx=1&sn=0de84df032b528dc04bc6f6a66204096&chksm=fc1b7588cb6cfc9e6c5f458da302f71dbf6aba5badbd70c8f4b63139fe7221f6ddf914c0ba1a&scene=21#wechat_re