1. 前言

上一篇文章中,我们聊到使用 xlrd、xlwt、xlutils 这一组合操作 Excel 的方法

最全总结 | 聊聊 Python 办公自动化之 Excel(上)

​本篇文章将继续聊另外一种方式,即:openpyxl

不得不说,openpyxl 更强大!

它支持 xlsx 格式的表格文件,并且支持 Numpy、Pandas 等包,可用于绘制图表

2. 准备

首先,我们需要安装依赖包

  1. # 安装依赖包
  2. pip3 install openpyxl

3. 读取数据

使用 openpyxl 中的 load_workbook(filepath) 加载本地一个 Excel 文件,返回结果是一个工作簿对象

  1. import openpyxl
  2. # 加载本地的Excel文件
  3. wb = openpyxl.load_workbook(file_path)

利用工作簿对象,可以获取所有的 Sheet 名称及 Sheet 列表

  1. def get_all_sheet_names(wb):
  2. """
  3. 获取所有sheet的名称
  4. :param wb:
  5. :return:
  6. """
  7. # sheet名称列表
  8. sheet_names = wb.sheetnames
  9. return sheet_names
  10. def get_all_sheet(wb):
  11. """
  12. 获取所有的sheet
  13. :param wb:
  14. :return:
  15. """
  16. # sheet名称列表
  17. sheet_names = get_all_sheet_names(wb)
  18. # 所有sheet
  19. sheets = []
  20. for sheet_name in sheet_names:
  21. sheet = wb[sheet_name]
  22. sheets.append(sheet)
  23. return sheets

工作簿对象提供了 active 属性,用于快速获取当前选择的 Sheet

  1. def get_current_sheet(wb):
  2. """
  3. 获取当前选择的sheet,默认是最后一个sheet
  4. :param wb:
  5. :return:
  6. """
  7. # 当前选中的sheet
  8. current_sheet = wb.active
  9. return current_sheet

另外,也可以通过 Sheet 名称去获取某一个特定的 Sheet 对象

  1. def get_sheet_by_name(wb, sheet_name):
  2. """
  3. 通过sheetname去查找某一个sheet
  4. :param wb:
  5. ​ :param sheet_name:
  6. :return:
  7. """
  8. sheet_names = get_all_sheet_names(wb)
  9. if sheet_name in sheet_names:
  10. result = wb[sheet_name]
  11. else:
  12. result = None
  13. return result

使用 sheet.max_row 和 sheet.max_column 可以获取当前 Sheet 中的数据行数和列数

  1. def get_row_and_column_num(sheet):
  2. """
  3. 获取sheet的行数和列数
  4. :param sheet:
  5. :return:
  6. """
  7. # 行数
  8. row_count = sheet.max_row
  9. # 列数
  10. column_count = sheet.max_column
  11. return row_count, column_count
  12. # 行数和列数
  13. row_count, column_count = get_row_and_column_num(sheet)
  14. print('行数和列数分别为:', row_count, column_count)

openpyxl 提供 2 种方式来定位一个单元格,分别是:

  • 数字索引,从 1 开始

    数字索引:行数字索引、列数字索引

    比如:row_index=1,column_index=1

  • 行和列组成的字符串索引

    字符串索引:列由字母组成 + 行索引

    比如:A1 对应第一行、第一列的单元格

并且,openpyxl.utils 提供了方法,便于 列索引 在两者之间进行转换

  1. from openpyxl.utils import get_column_letter, column_index_from_string
  2. def column_num_to_str(num):
  3. """
  4. Excel索引列从数字转为字母
  5. :param num:
  6. :return:
  7. """
  8. return get_column_letter(num)
  9. def column_str_to_num(str):
  10. """
  11. Excel索引列,从字母转为数字
  12. :param str:
  13. :return:
  14. """
  15. return column_index_from_string(str)

单元格的获取,同样可以通过上面 2 种索引方式来获取

  1. def get_cell(sheet, row_index, column_index):
  2. """
  3. 获取单元格
  4. :param sheet:
  5. :param row_index:
  6. :param column_index:
  7. :return:
  8. """
  9. # openpyxl索引都是从1开始计数,这与xlrd有所不同
  10. # 获取某一个单元格(二选一)
  11. # 比如:获取A1单元格的数据,即第一个行、第一列的数据
  12. # cell_one = sheet['A1']
  13. cell_one = sheet.cell(row=row_index, column=column_index)
  14. return cell_one

在日常处理 Excel 数据过程中,可能需要判断单元格数据类型,而 openpyxl 并没有提供现成的方法

这里,我们可以通过单元格对象的 value 属性拿到值,接着使用 isinstance 方法判断数据类型

  1. def get_cell_value_and_type(cell):
  2. """
  3. 获取某一个cell的内容及数据类型
  4. :param cell:
  5. :return:
  6. """
  7. # 单元格的值
  8. cell_value = cell.value
  9. # 单元格的类型
  10. cell_type = get_cell_value_type(cell_value)
  11. return cell_value, cell_type
  12. def get_cell_value_type(cell_value):
  13. """
  14. 获取数据类型
  15. :param cell_value:
  16. :return:
  17. """
  18. # 其中
  19. # 0:空
  20. # 1:数字
  21. # 2:字符串
  22. # 3:日期
  23. # 4:其他
  24. if not cell_value:
  25. cell_type = 0
  26. elif isinstance(cell_value, int) or isinstance(cell_value, float):
  27. cell_type = 1
  28. elif isinstance(cell_value, str):
  29. cell_type = 2
  30. elif isinstance(cell_value, datetime.datetime):
  31. cell_type = 3
  32. else:
  33. cell_type = 4
  34. return cell_type

单独获取某一行[列]的数据,可以使用下面的方式:

  1. def get_row_cells_by_index(sheet, row_index):
  2. """
  3. 通过行索引,获取某一行的单元格
  4. :param row_index:
  5. :return:
  6. """
  7. # 注意:第一列从1开始
  8. row_cells = sheet[row_index]
  9. return row_cells
  10. def get_column_cells_by_index(sheet, column_index):
  11. """
  12. 通过列索引,获取某一列的单元格
  13. """
  14. # 数字转为字母
  15. column_index_str = column_num_to_str(column_index)
  16. # 获取某一列的数据
  17. column_cells = sheet[column_index_str]
  18. return column_cells

需要注意的是,获取某一行的数据需要传入数字索引;而对于列数据的获取,必须传入字符串索引

和 Python 列表范围取值类似,openpyxl 同样支持使用 : 符号拿到某个范围内的数据行[列]

  1. def get_rows_by_range(sheet, row_index_start, row_index_end):
  2. """
  3. 通过范围去选择行范围
  4. 比如:选择第2行到第4行的所有数据,返回值为元组
  5. :param sheet:
  6. :param row_index_start:
  7. :param row_index_end:
  8. :return:
  9. """
  10. rows_range = sheet[row_index_start:row_index_end]
  11. return rows_range
  12. def get_columns_by_range(sheet, column_index_start, column_index_end):
  13. """
  14. 通过范围去选择列范围
  15. 比如:选择第2列到第4列的所有数据,返回值为元组
  16. :param sheet:
  17. :param column_index_start:
  18. :param column_index_end:
  19. :return:
  20. """
  21. columns_range = sheet[column_num_to_str(column_index_start):column_num_to_str(column_index_end)]
  22. return columns_range

4. 写入数据

要写入数据到 Excel 表格

首先,使用 openpyxl.Workbook() 创建一个 Excel 工作簿对象

接着,使用工作簿对象的 create_sheet() 新建一个 Sheet

  1. # 创建一个Excel工作簿
  2. # 注意:每次新建一个Excel文件,都会默认生成一个名称为【Sheet】的工作表Sheet
  3. wb = openpyxl.Workbook()
  4. # 创建一个新的sheet,默认被插到尾部
  5. # new_sheet = wb.create_sheet('新的Sheet')
  6. # 也可以通过第二个参数:index来指定插入的位置
  7. # 比如:插入到开头
  8. new_sheet = wb.create_sheet('新的Sheet', 0)

默认创建的 Sheet 被插入到最后一个位置,第 2 个参数可以指定 Sheet 插入的位置

Sheet 标签的背景色同样支持修改,使用 sheet_properties.tabColor 指定 RGB 颜色值

比如,要设置某一个 Sheet 的背景色为红色,只需要先查询到对应的 Sheet,然后指定颜色值为 FF0000 即可

  1. def set_sheet_bg_color(sheet, rgb_value):
  2. """
  3. 设置Sheet标签的颜色
  4. :param rgb_value:
  5. :return:
  6. """
  7. # 设置Sheet底部按钮的颜色(RRGGBB)
  8. sheet.sheet_properties.tabColor = rgb_value
  9. # 设置Sheet的背景色(红色)
  10. set_sheet_bg_color(new_sheet, 'FF0000')

openpyxl 支持行列数字索引、字符串索引以这 2 种方式写入数据到单元格中

  1. def write_value_to_cell_with_num(sheet, row_index, column_index, value):
  2. """
  3. 按行索引、列索引写入数据
  4. :param shell:
  5. :param row_index: 行索引
  6. :param column_index: 列索引
  7. :param value:
  8. :return:
  9. """
  10. # 二选一
  11. sheet.cell(row=row_index, column=column_index, value=value)
  12. # shell.cell(row=row_index, column=column_index).value = value
  13. def write_value_to_cell_with_index_str(sheet, index_str, value):
  14. """
  15. 按字母位置,写入数据到对应单元格
  16. :param shell:
  17. :param index_str: 字母对应的单元格位置
  18. :param value:
  19. :return:
  20. """
  21. sheet[index_str] = value

在单元格中插入图片也很简单,openpyxl 提供的 add_image() 方法

参数有 2 个,分别是:图片对象、单元格字符串索引

为了便于使用,我们可以将列索引进行转换,然后封装成两个插入图片的方法

  1. from openpyxl.drawing.image import Image
  2. def insert_img_to_cell_with_num(sheet, image_path, row_index, column_index):
  3. """
  4. 往单元格中插入图片
  5. :param sheet:
  6. :param image_path:
  7. :param row_index:
  8. :param column_index:
  9. :return:
  10. """
  11. # 通过行索引、列索引,获取到字母索引
  12. index_str = column_num_to_str(column_index) + str(row_index)
  13. insert_img_to_cell_with_str(sheet, image_path, index_str)
  14. def insert_img_to_cell_with_str(sheet, image_path, index_str):
  15. """
  16. 往单元格中插入图片
  17. :param sheet:
  18. :param image_path:
  19. :param index_str:
  20. :return:
  21. """
  22. sheet.add_image((image_path), index_str)

最后,调用工作簿对象的 save() 方法,将数据真实写入到 Excel 文件中

  1. # 注意:必须要写入,才能真实的保存到文件中
  2. wb.template = False
  3. wb.save('new.xlsx')

5. 修改数据

修改数据包含:单元格数据的修改、单元格样式的修改

对于单元格数据的修改,只需要先读取工作簿对象,查询到要操作的 Sheet 对象,然后调用上面的方法修改单元格数据,最后调用 save() 函数保存覆盖即可

  1. def modify_excel(self, file_path):
  2. """
  3. 修改本地Excel文件中数据
  4. :param file_path:
  5. :return:
  6. """
  7. # 读取本地Excel文件
  8. wb = openpyxl.load_workbook(file_path)
  9. # 读取某一个sheet
  10. sheet = wb['第一个Sheet']
  11. print(sheet)
  12. # 直接修改某一个单元格的数据
  13. write_value_to_cell_with_num(sheet, 1, 1, '姓名1')
  14. # 保存并覆盖
  15. wb.save(file_path)

单元格样式包含:字体样式、单元格背景样式、边框样式、对齐方式等

以常见的字体样式、对齐方式为例

首先,使用 openpyxl 中的 Font 类创建一个对象,指定字体名称、字体大小、是否加粗、是否斜体、颜色、下划线等

  1. from openpyxl.styles import Font
  2. # 字体格式
  3. # 指定字体类型、大小、是否加粗、颜色等
  4. font0 = Font(name='Calibri',
  5. size=20,
  6. bold=False,
  7. italic=False,
  8. vertAlign=None,
  9. underline='none',
  10. strike=False,
  11. color='FF00FF00')

接着,构建一个 Alignment 对象,指定单元格的对齐方式

  1. from openpyxl.styles import Font,Alignment
  2. # 单元格对齐方式
  3. alignment0 = Alignment(horizontal='center',
  4. vertical='bottom',
  5. text_rotation=0,
  6. wrap_text=False,
  7. shrink_to_fit=False,
  8. indent=0)

最后,使用单元格对象的 font/alignment 属性,将字体样式和对齐方式设置进去即可

  1. # 设置属性样式(字体、对齐方式)
  2. sheet['A1'].font = font0
  3. sheet['A1'].alignment = alignment0

6. 进阶用法

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

1、获取可见及隐藏的 Sheet

通过判断 Sheet 对象的 sheet_state 属性值,可以判断当前 Sheet 是显示还是隐藏

当值为 visible 时,代表 Sheet 是显示的

当值是 hidden 时,代表这个 Sheet 被隐藏了

  1. def get_all_visiable_sheets(wb):
  2. """
  3. 获取工作簿中所有可见的sheet
  4. :param wb:
  5. :return:
  6. """
  7. return [sheet for sheet in get_all_sheet(wb) if sheet.sheet_state == 'visible']
  8. def get_all_hidden_sheets(wb):
  9. """
  10. 获取工作簿中所有隐藏的sheet
  11. :param wb:
  12. :return:
  13. """
  14. return [sheet for sheet in get_all_sheet(wb) if sheet.sheet_state == 'hidden']

2、获取隐藏/显示的行索引列表、列索引列表

受限于篇幅,这里以获取所有显示/隐藏的行索引列表为例

遍历 Sheet 对象的 row_dimensions 属性值,通过判断行属性的 hidden 值,判断当前行是否隐藏或显示

  1. def get_all_rows_index(sheet, hidden_or_visiable):
  2. """
  3. 获取所有隐藏/显示的行
  4. :param hidden_or_visiable: True:隐藏;False:显示
  5. :param sheet:
  6. :return:
  7. """
  8. # 遍历行
  9. # 隐藏的索引
  10. hidden_indexs = []
  11. # 所有隐藏的行索引
  12. for row_index, rowDimension in sheet.row_dimensions.items():
  13. if rowDimension.hidden:
  14. hidden_indexs.append(row_index)
  15. # 所有显示的行索引
  16. visiable_indexs = [index + 1 for index in range(get_row_and_column_num(sheet)[0]) if index + 1 not in hidden_indexs]
  17. # 隐藏或者显示的行索引列表
  18. return hidden_indexs if hidden_or_visiable else visiable_indexs

3、获取单元格字体颜色及单元格背景颜色

单元格对象的 font.color.rgb、fill.fgColor.rgb 属性值分别代表字体颜色值、单元格背景颜色

  1. def get_cell_font_color(sheet, row_index, column_index):
  2. """
  3. 获取单元格字体的颜色
  4. :param sheet:
  5. :param row_index:行索引
  6. :param column_index:列索引
  7. :return:
  8. """
  9. cell_color = sheet.cell(row_index, column_index).font.color
  10. if cell_color:
  11. return sheet.cell(row_index, column_index).font.color.rgb
  12. else:
  13. # 颜色不存在,可能单元格没有数据
  14. return None
  15. def get_cell_bg_color(sheet, row_index, column_index):
  16. """
  17. 获取单元格背景的颜色
  18. :param sheet:
  19. :param row_index:行索引
  20. :param column_index:列索引
  21. :return:
  22. """
  23. return sheet.cell(row_index, column_index).fill.fgColor.rgb

7. 最后

可以发现,openpyxl 相比 xlrd/xlwt,提供了大量实用的 API,功能更强大,并且完美支持 xlsx!

受限于篇幅,文中只展示了部分功能和代码,更加复杂的功能,比如:单元格合并、单元格完整样式操作,我已经封装成方法上传到后台

要获取全部源码,关注公众号「 AirPython 」,后台回复「 excel 」即可获得全部源码

如果你觉得文章还不错,请大家 点赞、分享、留言下,因为这将是我持续输出更多优质文章的最强动力!

推荐阅读

最全总结 | 聊聊 Python 办公自动化之 Excel(上)

最全总结 | 聊聊 Python 数据处理全家桶(配置篇)

我用几行 Python 自动化脚本完美解决掉了小姐姐的微信焦虑感

最全总结 | 聊聊 Python 办公自动化之 Excel(中)的更多相关文章

  1. 最全总结 | 聊聊 Python 办公自动化之 Excel(下)

    1. 前言 前面谈到 Python 处理 Excel 文件最常见的两种方式,即:xlrd/xlwt.openpyxl ​其中, xlrd/xlwt 这一组合,xlrd 可以负责读取数据,而 xlwt ...

  2. 最全总结 | 聊聊 Python 办公自动化之 Excel(上)

    1. 前言 在我们日常工作中,经常会使用 Word.Excel.PPT.PDF 等办公软件 但是,经常会遇到一些重复繁琐的事情,这时候手工操作显得效率极其低下:通过 Python 实现办公自动化变的很 ...

  3. 最全总结 | 聊聊 Python 办公自动化之 Word(上)

    1. 前言 日常自动化办公中,使用 Python 真的能做到事半功倍! 在上一个系列中,我们对 Python 操作 Excel 进行了一次全面总结 最全总结 | 聊聊 Python 办公自动化之 Ex ...

  4. 最全总结 | 聊聊 Python 办公自动化之 Word(中)

    1. 前言 上一篇文章,对 Word 写入数据的一些常见操作进行了总结 最全总结 | 聊聊 Python 办公自动化之 Word(上) 相比写入数据,读取数据同样很实用! 本篇文章,将谈谈如何全面读取 ...

  5. 最全总结 | 聊聊 Python 办公自动化之 Word(下)

    1. 前言 关于 Word 文档的读写,前面两篇文章分别进行了一次全面的总结 最全总结 | 聊聊 Python 办公自动化之 Word(上) 最全总结 | 聊聊 Python 办公自动化之 Word( ...

  6. 最全总结 | 聊聊 Python 办公自动化之 PDF(上)

    1. 前言 自动化办公,非 Python 莫属! 从本篇文章开始,我们继续聊聊自动化办公中另外一个常用系列:PPT 2. 准备一下 Python 操作 PPT 最强大的依赖库是:python-pptx ...

  7. 最全总结 | 聊聊 Python 办公自动化之 PPT(中)

    1. 前言 上一篇文章简单地介绍了 PPT 的文档结构,并使用 python-pptx 这个依赖库完成对 PPT 文档最基本的操作 最全总结 | 聊聊 Python 办公自动化之 PPT(上) 作为 ...

  8. 最全总结 | 聊聊 Python 数据处理全家桶(PgSQL篇)

    1. 前言 大家好,我是安果! Python 数据处理全家桶,截止到现在,一共写过 6 篇文章,有兴趣的小伙伴可以去了解一下! 最全总结 | 聊聊 Python 数据处理全家桶(Mysql 篇) 最全 ...

  9. Python批量修改Excel中的文件内容

    import osimport xlrdfrom xlutils.copy import copydef base_dir(filename=None):    return os.path.join ...

随机推荐

  1. shell-的特殊变量-位置变量$0 $n $* $# $@详解

    一:shell特殊变量  1. 位置变量 $0 获取当前执行的shell脚本的文件名,包括路径 $n 获取当前执行的shell脚本的第n个参数值,n=1..9,当n为0时表示脚本的文件名,如果n大于9 ...

  2. 天啦噜!知道硬盘很慢,但没想到比 CPU L1 Cache 慢 10000000 倍

    前言 大家如果想自己组装电脑的话,肯定需要购买一个 CPU,但是存储器方面的设备,分类比较多,那我们肯定不能只买一种存储器,比如你除了要买内存,还要买硬盘,而针对硬盘我们还可以选择是固态硬盘还是机械硬 ...

  3. Rust之路(2)——数据类型 上篇

    [未经书面同意,严禁转载] -- 2020-10-13 -- Rust是系统编程语言.什么意思呢?其主要领域是编写贴近操作系统的软件,文件操作.办公工具.网络系统,日常用的各种客户端.浏览器.记事本. ...

  4. Android开发签名证书的生成

    现在都说互联网寒冬,其实只要自身技术能力够强,咱们就不怕!我这边专门针对Android开发工程师整理了一套[Android进阶学习视频].[全套Android面试秘籍].[Android知识点PDF] ...

  5. unix socket接口

    socket 创建套接字文件: #include <sys/socket.h> // 成功返回非负套接字描述符,失败返回-1 int socket(int domain, int type ...

  6. Jmeter请求之接口串联自动化测试(未完)

    方案一:添加Cookie管理器,把用户的登录状态存在cookie管理器中,类似于浏览器 存储测试结果: 监听器->保存响应到文件,对结果进行存储 文件名前缀:保存到哪个地方前缀是什么D:\tes ...

  7. spring boot: 用thymeleaf嵌套循环展示多层数据(spring boot 2.3.2)

    一,什么情况下会用到嵌套循环? 当我们展示多个分类时,每个分类下又展示出推荐的前几个商品,   这时我们需要用到嵌套循环 看一个例子: 说明:刘宏缔的架构森林是一个专注架构的博客,地址:https:/ ...

  8. spring boot:解决cors跨域问题的两种方法(spring boot 2.3.2)

    一,什么是CORS? 1,CORS(跨域资源共享)(CORS,Cross-origin resource sharing), 它是一个 W3C 标准中浏览器技术的规范, 它允许浏览器向非同一个域的服务 ...

  9. selenium---输入内容后搜索

    from time import sleep from selenium import webdriver br = webdriver.Chrome() url = "https://ww ...

  10. PowerShell 定时输出citrix desktop昨日连接会话

    asnp citrix*$now_date = [datetime]::Today$day2 = [datetime]::Today - [TimeSpan]::FromHours(24)$fiel_ ...