聊聊 Python 办公自动化之一 Excel

开发 后端 自动化
在我们日常工作中,经常会使用 Word、Excel、PPT、PDF 等办公软件,但是,经常会遇到一些重复繁琐的事情,这时候手工操作显得效率极其低下;通过 Python 实现办公自动化变的很有必要,接下来的 一系列 文章,我将带大家对 Python 办公自动化做一个全面的总结,绝对的干货!

 [[350611]]

本文转载自微信公众号「AirPython」,作者星安果。转载本文请联系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 文件

我们安装这两个依赖库

  1. # 安装依赖库 
  2. pip3 install xlrd  
  3. pip3 install xlwt  

3. xlrd 读取 Excel

使用 xlrd 中的 open_workbook(filepath) 打开本地一个 Excel 文件

  1. import xlrd 
  2.  
  3. # 打开文件,返回一个工作簿对象 
  4. wb = xlrd.open_workbook(file_path) 

工作簿对象的 nsheets 属性获取 Sheet 数目,sheet_names() 方法返回所有 Sheet 名称的列表

  1. # 统计sheet数量 
  2. sheets_num, sheets_names = wb.nsheets, wb.sheet_names() 
  3. print('sheet数量一共有:', sheets_num) 
  4. print('sheet名称分别为:', sheets_names) 

筛选出工作簿中的某一个 Sheet 有 2 种方式,分别是:

  • 通过 Sheet 名称
  • 使用位置索引,从 0 开始
  1. # 获取某一个sheet 
  2. # 通过名称或者索引获取 
  3. sheet = wb.sheet_by_index(0) 
  4.  
  5. # sheet = wb.sheet_by_name('第一个Sheet'
  6. print(sheet) 

每一个 sheet 对象都可以利用 name、nrows、ncols 获取 Sheet 名称、行数量、列数量

另外

row_values(index)、col_values(index) 分别用于获取某一行或某一列的数据列表

  1. # 获取某一个sheet中,包含的行数量、列数量 
  2. sheet_name, sheet_row_count, sheet_column_count = sheet.name, sheet.nrows, sheet.ncols 
  3. print('当前sheet名称为:', sheet_name, ",一共有:", sheet_row_count, "行;有:", sheet_column_count, "列"
  4.  
  5. # 单独获取某一行数据,索引从0开始 
  6. # 比如:获取第2行数据 
  7. row_datas = sheet.row_values(1) 
  8. print('第2行数据为:', row_datas) 
  9.  
  10. # 单独获取某一列数据,索引从0开始 
  11. # 比如:获取第二列数据 
  12. column_datas = sheet.col_values(1) 
  13. 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(错误)
  1. # 获取某一个单元格的数据 
  2. # 比如:获取第2行第1列的单元格的数据 
  3. one_cell = sheet.cell(1, 0) 
  4. # 单元格的值 
  5. cell_value = one_cell.value 
  6. print("单元格的值为:", cell_value) 
  7. # 单元格数据类型 
  8. cell_type = one_cell. 
  9. print("单元格数据类型为:", cell_type) 

最后,如果要获取当前 Sheet 所有单元格中的数据,可以通过遍历所有行、列来操作

  1. # 获取所有单元格的值 
  2. print('表格中所有数据如下:'
  3. for r in range(sheet.nrows): 
  4.     for i in range(sheet.ncols): 
  5.         print(sheet.cell(r, i).value) 

4. xlwt 写入 Excel

如果想实现将数据写入到 Excel 中,xlwt 就很方便了

首先,使用 xlwt 的 Workbook() 方法创建一个工作簿对象

然后,使用工作簿对象的 add_sheet(sheetname) 方法新增 Sheet

  1. import xlwt 
  2.  
  3. sheetname = '第一个Sheet' 
  4.  
  5. # 创建一个工作簿对象 
  6. wb = xlwt.Workbook() 
  7.  
  8. # 添加Sheet,通过sheet名称 
  9. sheet = wb.add_sheet(sheetname) 

接着,通过 sheet 对象的 write() 方法,按照行索引和列索引,将数据写入到对应单元格中去

  1. # 将数据写入到Sheet中 
  2. # 3个参数分别是:行索引(从0开始)、列索引(从0开始)、单元格的值 
  3. # 第一行第一列,写入一个数据 
  4. # 写入标题 
  5. for index, title in enumerate(self.titles): 
  6.     sheet.write(0, index, title) 
  7.  
  8. # 写入值 
  9. for index_row, row_values in enumerate(self.values): 
  10.     for index_column, column_value in enumerate(row_values): 
  11.         sheet.write(index_row + 1, index_column, column_value) 

需要注意的是,最后必须调用工作簿的 save(filepath),才能在本地生成 Excel 文件

  1. # 保存文件 
  2. # 最后保存文件即可 
  3. wb.save(filepath) 

5. 进阶用法

接下来,聊聊几个常用的进阶用法

1、获取所有可见的 Sheet

在读取 Sheet 数据时,经常需要过滤隐藏的 Sheet

当 sheet 对象的 visibility 属性值为 0 时,代表此 Sheet 在工作簿中是显示的;否则被隐藏了

  1. def get_all_visiable_sheets(self, wb): 
  2.     ""
  3.     获取所有可见的sheet 
  4.     :param wb: 
  5.     :return
  6.     ""
  7.     return list(filter(lambda item: item.visibility == 0, wb.sheets())) 
  8.  
  9. # 1、获取所有可看见的sheet 
  10. sheet_visiable = self.get_all_visiable_sheets(wb) 
  11. print('所有可见的sheet包含:', sheet_visiable) 

2、获取 Sheet 可见行或列

某一个 Sheet 中,可能存在部分行、列被隐藏了

  1. def get_all_visiable_rows(self, sheet): 
  2.     ""
  3.     获取某一个sheet中,可见的行 
  4.     :param sheet: 
  5.     :return
  6.     ""
  7.     result = [index for index in range(sheet.nrows) if sheet.rowinfo_map[index].hidden == 0] 
  8.     return result 
  9.  
  10. def get_all_visiable_columns(self, sheet): 
  11.     ""
  12.     获取某一个sheet中,可见的列 
  13.     :param sheet: 
  14.     :return
  15.     ""
  16.     result = [index for index in range(sheet.ncols) if sheet.colinfo_map[index].hidden == 0] 
  17.     return result 

3、获取单元格的样式

以获取单元格字体颜色和背景为例

  1. def get_cell_bg_color(self, wb, sheet, row_index, col_index): 
  2.     ""
  3.     获取某一个单元格的背景颜色 
  4.     :param wb: 
  5.     :param sheet: 
  6.     :param row_index: 
  7.     :param col_index: 
  8.     :return
  9.     ""
  10.     xfx = sheet.cell_xf_index(row_index, col_index) 
  11.     xf = wb.xf_list[xfx] 
  12.  
  13.     # 字体颜色 
  14.     font_color = wb.font_list[xf.font_index].colour_index 
  15.     # 背景颜色 
  16.     bg_color = xf.background.pattern_colour_index 
  17.  
  18.     return font_color, bg_color 

需要注意的是,使用 xlrd 读取单元格的样式,打开工作簿的时候需要显式定义 formatting_info = True,否则会抛出异常

  1. # 注意:必须设置formatting_info=True,才能正常获取属性 
  2. wb = xlrd.open_workbook(file_path, formatting_info=True
  3. sheet = wb.sheet_by_index(0 

6. 最后搭配使用 xlrd、xlwt,基本上能完成大部分的工作,对于一些复杂的功能,比如:复制、分割、筛选等功能,可以用上 xlutils 这个依赖库

需要指出的是,这个组合对 xlsx 的兼容性不太好;如果需要操作 xlsx 文件,需要先转为 xls,然后再进行

责任编辑:武晓燕 来源: AirPython
相关推荐

2020-04-21 10:45:47

PythonWordExcel

2022-06-13 09:12:04

PythonPDF代码

2022-08-04 13:27:35

Pythonopenpyxl

2022-03-21 10:09:08

PythonExcel邮件

2023-05-15 08:01:32

Python办公自动化

2021-04-22 15:46:42

物联网智能办公IoT

2009-02-11 09:49:00

网络维护网络管理

2024-05-29 11:16:33

PythonExcel

2022-06-05 21:09:47

Python办公自动化

2009-10-26 21:26:37

网络运维办公自动化

2011-08-03 17:37:05

2021-07-09 07:24:47

Python自动化办公

2013-01-10 15:21:09

三元食品办公自动化IBM

2013-12-09 18:12:02

华为PTN电力配网

2021-12-28 09:24:49

Python邮件Word

2021-04-23 22:44:57

Python开发办公自动化

2021-02-01 12:24:40

Python日志Expect

2023-10-18 13:57:17

2023-11-20 22:07:51

PythonPDF

2022-07-05 08:26:10

Python报表自动化邮箱
点赞
收藏

51CTO技术栈公众号