公司工资条经常使用Excel制作,但是每个月都要做一遍,能不能用python写个程序自动化完成这想工作?当然可以,而且只是分分钟的事!
先来看看原始数据是什么样子:
最后做成的效果:
使用Excel每次都需要手动修改一遍,对python来说是很简单的,15行代码就可以秒出一个工资条。
一、操作逻辑
逻辑本身非常简单,其实就是在每行之间插入一个空行和一个表头行
二、代码模块
首先是读取Excel表格,读取表格所有内容,python读取excel的方式有很多,最简洁的就是pandas,pandas可以通过read_excel()和to_excel()非常简单的读写excel。读取数据一行搞定:
import pandas
excel_data = pandas.read_excel('E:\\python_practice\\工资表.xlsx') # 读取信息
print(excel_data) # 输出结果
序号 姓名 部门 职务 基本工资 工龄津贴 岗位补贴 奖金 应发 五险一金 实发
0 1 张三 人事部 经理 8600 700 300 5000 14600 1800 12800
1 2 李四 行政部 副经理 8600 700 300 3000 12600 1800 10800
2 3 王五 市场部 经理 8600 700 300 5000 14600 1800 12800
3 4 赵六 市场部 职员 5600 300 100 1000 7000 1300 5700
4 5 孙七 市场部 职员 5600 300 100 1000 7000 1300 5700
插入行其实只是我们人为的动作,python是可以单独插入来实现,但是我们可以逐行写入文件,直接在写文件时多写两行即可,省去了操作数据插入的过程(实际上即便单独插入用pandas也是秒完成的,能少一步就不多一步),只是涉及到逐行插入,逐行写入to_excel()是不行的,可以使用openoyxl,好的,我们先来把重复插入的两行创建出来:
excel_head = list(excel_data.columns)
excel_null = ['', '', '', '', '', '', '', '', '', '', '']
导入 openpyxl,并创建对象
import openpyxl
wb = openpyxl.Workbook() # 创建一个对象
sheet = wb.active
三、循环写入文件
使用双层for循环写数据到文件,只是这里思维逻辑上有一点比较难理解,就是新文件的行序号和源数据的行序号的对应关系,这涉及到取数据赋值,代码如下:
row = 0
for r in range(0, excel_data.shape[0] * 3, 3):
for c in range(excel_data.shape[1]):
sheet.cell(r + 1, c + 1, value=excel_head[c])
sheet.cell(r + 2, c + 1, value=excel_data.iloc[row, c])
sheet.cell(r + 3, c + 1, value=excel_null[c])
row += 1
四、完整代码
别看上面说了很多,实际代码只需要15行!
import pandas
import openpyxl
excel_data = pandas.read_excel('E:\\青岛慧涵信息科技有限公司\\工资表.xlsx') # 读取信息
excel_head = list(excel_data.columns)
excel_null = ['', '', '', '', '', '', '', '', '', '', '']
wb = openpyxl.Workbook() # 创建一个对象
sheet = wb.active
row = 0
for r in range(0, excel_data.shape[0] * 3, 3):
for c in range(excel_data.shape[1]):
sheet.cell(r + 1, c + 1, value=excel_head[c])
sheet.cell(r + 2, c + 1, value=excel_data.iloc[row, c])
sheet.cell(r + 3, c + 1, value=excel_null[c])
row += 1
wb.save('E:\\青岛慧涵信息科技有限公司\\工资表-xg.xlsx')
五、其他设置
目前处理完成的文件格式使用的时默认格式,并没有使用对齐方式,本身工资条也没必要太花哨的格式,如果需要,也不要紧,pandas也可以满足你任意格式设置!这里就不展开了。只是这是源代码,如果是给非IT人员使用,那就使用 pyinstall打包个exe文件,那都不是事!
原文链接:https://blog.csdn.net/wuwei_201/article/details/108040661