聊聊python 辦公自動化之 Excel（上）

做者：星安果

來源：AirPython

1. 前言

在咱們平常工做中，常常會使用 Word、Excel、PPT、PDF 等辦公軟件。可是，常常會遇到一些重複繁瑣的事情，這時候手工操做顯得效率極其低下；經過python 實現辦公自動化變的頗有必要。

接下來的 一系列 文章，我將帶你們對 Python 辦公自動化作一個全面的總結，絕對的乾貨！

2. 準備

使用 Python 操做 Excel 文件，常見的方式以下：

• xlrd / xlwt
• openpyxl
• Pandas
• xlsxwriter
• xlwings
• pywin32

xlrd 和 xlwt 是操做 Excel 文件最多的兩個依賴庫。其中，xlrd 負責讀取 Excel 文件，xlwt 能夠寫入數據到 Excel 文件，咱們安裝這兩個依賴庫。

安裝依賴庫

pip3 install xlrd

pip3 install xlwt

3. xlrd 讀取 Excel

使用 xlrd 中的 open_workbook(filepath) 打開本地一個 Excel 文件。

import xlrd

打開文件，返回一個工做簿對象

wb = xlrd.open_workbook(file_path)

工做簿對象的 nsheets 屬性獲取 Sheet 數目，sheet_names() 方法返回全部 Sheet 名稱的列表。

統計sheet數量

sheets_num, sheets_names = wb.nsheets, wb.sheet_names()

print('sheet數量一共有:', sheets_num)

print('sheet名稱分別爲:', sheets_names)

篩選出工做簿中的某一個 Sheet 有 2 種方式，分別是：

• 經過 Sheet 名稱
• 使用位置索引，從 0 開始

獲取某一個sheet

經過名稱或者索引獲取

sheet = wb.sheet_by_index(0)

sheet = wb.sheet_by_name('第一個Sheet')

print(sheet)

每個 sheet 對象均可以利用 name、nrows、ncols 獲取 Sheet 名稱、行數量、列數量。

另外，row_values(index)、col_values(index) 分別用於獲取某一行或某一列的數據列表。

獲取某一個sheet中，包含的行數量、列數量

sheet_name, sheet_row_count,

sheet_column_count = sheet.name, sheet.nrows, sheet.ncols

print('當前sheet名稱爲:',

sheet_name, ",一共有：",

sheet_row_count, "行；有：",

sheet_column_count, "列")

單獨獲取某一行數據，索引從0開始

好比：獲取第2行數據

row_datas = sheet.row_values(1)

print('第2行數據爲：', row_datas)

單獨獲取某一列數據，索引從0開始

好比：獲取第二列數據

column_datas = sheet.col_values(1)

print('第2列數據爲：', column_datas)

單元格能夠經過行索引、列索引，調用 cell(row_index,column_index) 函數獲取。須要注意的是，行索引和列索引都是從 0 開始，即：0 表明第一行。在 xlrd 中，單元格的數據類型包含 6 種，用 ctype 屬性對應關係以下：

• 0 -- 空（empty）
• 1 -- 字符串（string）
• 2 -- 數字（number）
• 3 -- date（日期）
• 4 -- boolean（布爾值）
• 5 -- error（錯誤）

獲取某一個單元格的數據

好比：獲取第2行第1列的單元格的數據

one_cell = sheet.cell(1, 0)

單元格的值

cell_value = one_cell.value

print("單元格的值爲:", cell_value)

單元格數據類型

cell_type = one_cell.

print("單元格數據類型爲:", cell_type)

最後，若是要獲取當前 Sheet 全部單元格中的數據，能夠經過遍歷全部行、列來操做。

獲取全部單元格的值

print('表格中全部數據以下：')

for r in range(sheet.nrows):

for i in range(sheet.ncols):

print(sheet.cell(r, i).value)

4. xlwt 寫入 Excel

若是想實現將數據寫入到 Excel 中，xlwt 就很方便了。

首先，使用 xlwt 的 Workbook() 方法建立一個工做簿對象；

而後，使用工做簿對象的 add_sheet(sheetname) 方法新增 Sheet；

import xlwt

sheetname = '第一個Sheet'

建立一個工做簿對象

wb = xlwt.Workbook()

添加Sheet，經過sheet名稱

sheet = wb.add_sheet(sheetname)

接着，經過 sheet 對象的 write() 方法，按照行索引和列索引，將數據寫入到對應單元格中去。

將數據寫入到Sheet中

3個參數分別是：行索引（從0開始）、列索引（從0開始）、單元格的值

第一行第一列，寫入一個數據

寫入標題

for index, title in enumerate(self.titles):

sheet.write(0, index, title)

寫入值

for index_row, row_values in enumerate(self.values):

for index_column, column_value in enumerate(row_values):

sheet.write(index_row + 1, index_column, column_value)

須要注意的是，最後必須調用工做簿的 save(filepath)，才能在本地生成 Excel 文件。

保存文件

最後保存文件便可

wb.save(filepath)

5. 進階用法

接下來，聊聊幾個經常使用的進階用法

一、獲取全部可見的 Sheet

在讀取 Sheet 數據時，常常須要過濾隱藏的 Sheet

當 sheet 對象的 visibility 屬性值爲 0 時，表明此 Sheet 在工做簿中是顯示的；不然被隱藏了

def get_all_visiable_sheets(self, wb):

"""

獲取全部可見的sheet

:param wb:

:return:

"""

return list(filter(lambda item: item.visibility == 0, wb.sheets()))

一、獲取全部可看見的sheet

sheet_visiable = self.get_all_visiable_sheets(wb)

print('全部可見的sheet包含：', sheet_visiable)

二、獲取 Sheet 可見行或列

某一個 Sheet 中，可能存在部分行、列被隱藏了。

def get_all_visiable_rows(self, sheet):

"""

獲取某一個sheet中，可見的行

:param sheet:

:return:

"""

result = [index for index in range(sheet.nrows) if sheet.rowinfo_map[index].hidden == 0]

return result

def get_all_visiable_columns(self, sheet):

"""

獲取某一個sheet中，可見的列

:param sheet:

:return:

"""

result = [index for index in range(sheet.ncols) if sheet.colinfo_map[index].hidden == 0]

return result

三、獲取單元格的樣式

以獲取單元格字體顏色和背景爲例。

def get_cell_bg_color(self, wb, sheet, row_index, col_index):

"""

獲取某一個單元格的背景顏色

:param wb:

:param sheet:

:param row_index:

:param col_index:

:return:

"""

xfx = sheet.cell_xf_index(row_index, col_index)

xf = wb.xf_list[xfx]

字體顏色

font_color = wb.font_list[xf.font_index].colour_index

背景顏色

bg_color = xf.background.pattern_colour_index

return font_color, bg_color

須要注意的是，使用 xlrd 讀取單元格的樣式，打開工做簿的時候須要顯式定義 formatting_info = True，不然會拋出異常。

注意：必須設置formatting_info=True，才能正常獲取屬性

wb = xlrd.open_workbook(file_path, formatting_info=True)

sheet = wb.sheet_by_index(0

6. 最後

搭配使用 xlrd、xlwt，基本上能完成大部分的工做，對於一些複雜的功能，好比：複製、分割、篩選等功能，能夠用上 xlutils 這個依賴庫。須要指出的是，這個組合對 xlsx 的兼容性不太好；若是須要操做 xlsx 文件，須要先轉爲 xls，而後再進行。