在工作中遇到需要将数据输出到excel,且需要对其中一些单元格进行合并,在网上找到了一些资料,在这里分享给大家。原文地址
比如如下表表格,需要根据A列的值,合并B,C列的对应单元格
2b26a0597eab4c01b21d9483e9b6ee69_image.png
pandas中的to_excel方法只能对索引进行合并,而xlsxwriter中,虽然提供有merge_range方法,但是这只是一个和基础的方法,每次都需要编写繁琐的测试才能最终调好,而且不能很好的重用。所以想自己写一个方法,结合dataframe和merge_range。大概思路是:
- 1,定义一个MY_DataFrame类,继承DataFrame类,这样能很好的利用pandas的很多特性,而不用自己重新组织数据结构。
- 2 ,定义一个my_mergewr_excel方法,参数分别为:输出excel的路径,用于判断是否需要合并的key_cols列表,用于指明哪些列上的单元格需要被合并的列表
- 3,将MY_DataFrame封装为一个My_Module模块,以备重用。
合并的算法如下:
- 1,根据给定参数的【关键列】,进行分组计数和排序,添加CN和RN两个辅助列
- 2,判断CN大于1的,该分组需要合并,否则该分组(行)无需合并(CN = 1说明这个分组数据行是唯一的,无需合并)
- 3,对应需要合并的分组,判断当前列是不是在给定参数【合并列】中,是则用合并写excel单元格