转载请注明:陈熹 chenx6542@foxmail.com (简书号:半为花间酒)
若公众号内转载请联系公众号:早起Python
本例可以学到的知识点:
openpyxl
模块的综合运用
练习数据:链接 https://pan.baidu.com/s/1o3xxsFO2KqozVbH68UwElg 提取码 r3r2
一、需求描述:
有一份 档案记录总表的Excel工作簿, 每天会根据 当天日期 建立新表,每天的表格内包含所有档案信息,同时也有可能会添加新的档案名。同个年度的总表在年末可能会有两、三百个工作表,同时每个表中可能也存在千余份档案信息。
(为了直观呈现本例以7个工作表和十余份档案的形式呈现)
表格形式如下:
需要完成的操作:为了方便审查特定档案信息,需要给出档案名后生成一份新表,该表包含指定档案在所有日期(即所有工作表)中的记录
最终结果如下(以 档案x003 为例):
二、步骤分析
正式写代码前可以把需求分析清楚,将复杂问题简单化
说白了,这个需求要求把所有日期工作表中的特定行都提取出来整合成一个新表。那么我们可以遍历每一张表,然后遍历第一列(名称列,也可以看作A列)每一个有数据的单元格,如果单元格中的文字为我们需要的档案名,就把这一行提取出来放到新的表格中
进一步梳理步骤为:
- 建立一个新的EXCEL工作簿
- 新表的表头和档案记录Excel中的一样,也是“名称”、“配置”、“提交日期”等
- 遍历档案记录Excel的每一张工作表sheet,再遍历第一列每一个有数据的单元格,对内容进行判断
- 找到符合条件的单元格后获取行号,根据行号将当前表中的特定行提取出来,并将行追加新创建的表中
分析清楚就着手写代码
三、代码实现
- 导入需要的库
本例中涉及旧表的打开和新表的创建,因此需要从openpyxl
导入load_workbook
和Workbook
(如果是ppt和word用到的模块就更智能了,一个方法就能搞定)
from openpyxl import load_workbook, Workbook
- 导入旧表及创建新表
# 从桌面上获取总表
filepath = r'C:\Users\chenx\Desktop\台账.xlsm' # 根据实际情况进行修改
workbook = load_workbook(filepath)
# 创建新的Excel工作簿获取到工作表
new_workbook = Workbook()
new_sheet = new_workbook.active
# 给新表写入表头
new_headers = ['名称', '配置', '提交日期', '受限操作', '操作时间', '状态', '存储位置']
new_sheet.append(new_headers)
- 核心步骤:多次遍历
可以用workbook.sheetnames
获取工作簿所有工作表名称的列表,然后遍历即可
for i in workbook.sheetnames:
sheet = workbook[i]
# 获取档案名称所在列
names = sheet['A']
按照前面的分析,需要遍历名称列,判断每一个单元格的值是不是需要的档案名。这里应注意,如果已经循环到需要的单元格,就可以停止循环了,但一定要把符合单元格的行号传递给一个变量做记录,不然一旦break
出循环就没有记忆了
flag = 0
for cell in names:
if cell.value == keyword: # 这里的keyword就是档案名,可以以 档案x003 为例
flag = cell.row
break
获得到符合条件的行号后用sheet[flag]
就可以拿到符合行了。openpyxl
不支持旧表的一整行写入新表,因此应对策略就是将这一行的所有单元格具体值组装成一个列表,用sheet.append(列表)
的方法写入新表
遍历部分的完整代码如下:
for i in workbook.sheetnames:
sheet = workbook[i]
names = sheet['A']
flag = 0
for cell in names:
if cell.value == keyword:
flag = cell.row
break
if flag: # 如果flag没有被修改则不需要顺序进行下列代码
data_lst = []
for cell in sheet[flag]:
# 这里加上一个对内容的判断,是让无内容的行直接放空,而不是写入一个 none
if cell.value:
data_lst.append(str(cell.value))
else:
data_lst.append(' ')
new_sheet.append(data_lst)
最后记得保存
new_workbook.save(r'C:\Users\chenx\Desktop\台账查询.xlsx')
写在最后
这是经过一定改编的真实案例,可见python自动化办公确实能够帮助我们解放自己的双手
如果你有正在烦恼的办公需求,欢迎在 早起Python 公众号后台留言
网友评论