自动化办公案例
目录
文件查找
import os
path='./'
filenames=os.listdir(path)
print(filenames)
文件读写
import os
# 设置文件夹路径,获取文件夹下的所有文件名
path = './工作文件夹/'
files_list = os.listdir(path)
# 设置需要查找的关键词
key_word = input("请输入要查找的关键词:")
# 打开结果文件
result_file = open('./demo_result.txt', 'a', encoding='utf-8')
# 循环处理每一个文件
for file_name in files_list:
# 判断文件类型是否在文件名中
if '.txt' in file_name:
# 找到文件时先打印提示
print("找到了文件:" + file_name)
# 将文件夹路径和文件名拼接成该文件的相对路径
target_file = path + file_name
# 打开文件,读取文件内容,然后关闭文件
file = open(target_file, 'r', encoding='utf-8')
content = file.read()
file.close()
# 判断关键词是否在文件内容中
if key_word in content:
# 匹配到关键词时先打印提示
print("妙啊,文件**{}**包含了关键词:{}".format(target_file, key_word))
# 将包含关键词的文档的文件路径,写入结果文件。
result_file.write(target_file + '\n')
# 关闭结果文件
result_file.close()
csv文件读写
import csv
# 设置员工发展基金确认表路径
source_path = './员工发展基金确认表.csv'
# 设置存放拆分结果文件的文件夹路径
result_path = './员工发展基金文件夹_demo/'
# 打开员工发展基金确认表
with open(source_path, 'r', encoding='utf-8', newline='') as source_file:
#读取文件一行一行,为列表
#csv_reader = csv.reader(csv_file)
# 将文件对象转换为DictReader对象
source_csv = csv.DictReader(source_file)
# 将csv对象的表头读取出来
headers = source_csv.fieldnames
# 循环处理确认表中除表头外的每一行数据
for csv_row in source_csv:
# 根据获取的员工名字拼接新文件名
file_name = csv_row['姓名'] + '.csv'
# 拼接新文件路径
file_path = result_path + file_name
# 创建新文件并添加数据
with open(file_path, 'w', encoding='utf-8', newline='') as target_file:
# 将文件对象转换为DictWriter对象
target_writer = csv.DictWriter(target_file, fieldnames=headers)
# 写入表头
target_writer.writeheader()
# 写入数据
target_writer.writerow(csv_row)
openpyxl的Excel文件操作
from openpyxl import load_workbook, Workbook
# 通过文件路径,打开已有工作簿
wb1 = load_workbook('./demo_excel.xlsx')
# 用 Workbook() 创建新工作簿
wb2 = Workbook()
# 新建工作簿
new_wb = Workbook()
# 打印工作簿对象
print(new_wb)
# 将新建的工作簿保存为【new_excel.xlsx】
new_wb.save('./new_excel.xlsx')
# 打开【公司人员名单.xlsx】工作簿
staff_wb = load_workbook('./codes/material/公司人员名单.xlsx')
# 获取活动工作表
active_ws = staff_wb.active
# 打印单元格对象A1
print(staff_ws['A1'])
# 修改单元格对象C2的值为10000
staff_ws['C2'].value = 10000
# 打印修改后的单元格对象C2的值
print(staff_ws['C2'].value)
# 打印获取到的第五行数据
print(active_ws[5])
# 打印获取到的第二列数据
print(active_ws['B'])
# 打印工作簿对象
print(staff_wb)
# 打印工作表对象
print(active_ws)
# for循环遍历,取出第三行的所有单元格对象
for row_cell in staff_ws[3]:
print(row_cell)
# for循环遍历,取出第三列(C列)的所有单元格对象
for col_cell in staff_ws['C']:
print(col_cell)
info_list = ['S1911', '萧爵瑟', 3000, '内容']
info_tuple = ('S1912', '吴琐薇', 5000, '销售')
#添加数据到最后行
active_ws.append(info_list)
active_ws.append(info_tuple)
# 保存工作簿为【append_demo.xlsx】
staff_wb.save('./append_demo.xlsx')
# 按表名取表
fhy_ws = staff_wb['上半年公司名单'] # fhy为first half year(上半年)的缩写
shy_ws = staff_wb['下半年公司名单'] # shy为second half year(下半年)的缩写
# 打印工作簿对象
print(staff_wb)
# 打印工作表对象
print(fhy_ws)
print(shy_ws)
# 返回第2行至第12行,第2列(B列)至第3列(C列)这个范围的单元格内的所有数据(值)
for row in fhy_ws.iter_rows(min_row=2, max_row=12, min_col=2, max_col=3,values_only=True):
print(row)
#values_only True返回每个单元格值的元组 ,False返回每个单元格对象的元组
excel表格读写
案例1 获取个人信息,并保存文件
from openpyxl import load_workbook
# 打开【10月员工绩效表】的工作簿,获取活动工作表
performance_wb = load_workbook('./material/10月员工绩效表.xlsx')
performance_ws = performance_wb.active
# 打开【江宇工资信息表】的工作簿,获取活动工作表
info_wb = load_workbook('./material/江宇工资信息表.xlsx')
info_ws = info_wb.active
# 获取【绩效】值
performance = performance_ws['D14'].value
# 获取【奖金】值
bonus = performance_ws['E14'].value
# 获取【基本工资】值
base = performance_ws['F14'].value
# 写入【绩效】值
info_ws['E11'].value = performance
# 写入【奖金】值
info_ws['F11'].value = bonus
# 写入【基本工资】值
info_ws['G11'].value = base
# 保存对【江宇工资信息表】工作簿的写入
info_wb.save('./material/江宇工资信息表.xlsx')
案例2 获取前十行数据并保存为新文件。
# 从openpyxl库导入load_workbook和Workbook
from openpyxl import load_workbook, Workbook
# 打开【10月员工绩效表.xlsx】工作簿
performance_wb = load_workbook('./material/10月员工绩效表.xlsx')
# 获取活动工作表
performance_ws = performance_wb.active
# 新建工作簿
new_wb = Workbook()
# 获取活动工作表
new_ws = new_wb.active
# 获取performance_ws的前十行数据
for row in performance_ws.iter_rows(max_row=10, values_only=True):
# 将数据写入新的工作表
new_ws.append(row)
# 保存新工作簿为【员工绩效表-模板.xlsx】
new_wb.save('员工绩效表-模板.xlsx')
案例3 计算并打印奖金信息
# 从openpyxl库导入load_workbook和Workbook
from openpyxl import load_workbook, Workbook
# 打开【10月员工绩效表.xlsx】工作簿
performance_wb = load_workbook('./material/10月员工绩效表.xlsx')
# 获取活动工作表
performance_ws = performance_wb.active
# 获取performance_ws中除表头外的数据
for row in performance_ws.iter_rows(min_row=2, values_only=True):
# 读取【工号】
staff_id = row[0]
# 读取【员工姓名】
staff_name = row[1]
# 读取【绩效】
performance = row[3]
# 读取【提成】
bonus = row[4]
# 计算“奖金”
award = performance + bonus
# 打印结果
print('工号:{},姓名:{},本月奖金为:{}'.format(staff_id, staff_name, award))
案例4 读取数据并在内存创建字典格式数据
# 从openpyxl库导入load_workbook函数
from openpyxl import load_workbook
# 打开【10月员工绩效表.xlsx】工作簿
performance_wb = load_workbook('./material/10月员工绩效表.xlsx')
# 获取活动工作表
performance_ws = performance_wb.active
# 创建员工信息字典
staff_info = {}
# 从第二行开始读取工作表中的信息
for row in performance_ws.iter_rows(min_row=2, values_only=True):
# 取出工号
member_number = row[0]
# 将信息存入员工信息字典
staff_info[member_number] = {
'工号': row[0],
'姓名': row[1],
'部门': row[2],
'绩效': row[3],
'奖金': row[4],
'基本工资': row[5],
'是否确认': row[6]}
print(staff_info)
案例5 取数汇总并写入
# 课后练习:取数汇总并写入
from openpyxl import load_workbook
performance_wb=load_workbook('./material/10月员工绩效表.xlsx')
performance_ws=performance_wb.active
info_wb=load_workbook('./material/江宇工资信息表.xlsx')
info_ws=info_wb.active
info_ws['H1'].value='总工资'
print(type(performance_ws['E11'].value))
print(type(performance_ws['F11'].value))
print((performance_ws['G11'].value))
info_ws['H11'].value='='+performance_ws['D14']+'+'+performance_ws['E14']+'+'+performance_ws['F14']
info_wb.save('./material/江宇工资信息表.xlsx')
案例6 数据筛选
from openpyxl import load_workbook, Workbook
# 打开【10月考勤统计.xlsx】工作簿
wb = load_workbook('./material/10月考勤统计.xlsx')
# 获取活动工作表
ws = wb.active
# 获取表头
late_header = []
for cell in ws[1]:
late_header.append(cell.value)
# 新建工作簿
new_wb = Workbook()
# 获取新工作簿中的工作表
new_ws = new_wb.active
# 将表头写入新工作簿的工作表中
new_ws.append(late_header)
# 从第二行开始遍历表格
for row in ws.iter_rows(min_row=2, values_only=True):
# 取出姓名,迟到时间和迟到次数
name = row[1]
time = row[3]
number = row[-1]
# 判断是否迟到
if time > 45 and number > 3:
print('{}迟到了{}分钟,迟到了{}次'.format(name, time, number))
# 将迟到人员信息写入新工作簿的工作表中
new_ws.append(row)
# 将新工作簿保存为【10月迟到人员信息.xlsx】
new_wb.save('./material/10月迟到人员信息.xlsx')
案例7 设置表格样式
import os
from openpyxl import load_workbook
from openpyxl.styles import PatternFill, Alignment, Side, Border
# 定义表头颜色样式为橙色
header_fill = PatternFill('solid', fgColor='FF7F24')
# 定义表中、表尾颜色样式为淡黄色
content_fill = PatternFill('solid', fgColor='FFFFE0')
# 定义表尾颜色样式为淡桔红色
bottom_fill = PatternFill('solid', fgColor='EE9572')
# 定义对齐样式横向居中、纵向居中(center,left,right,center,top,bottom)
align = Alignment(horizontal='center', vertical='center')
# 定义边样式为细条
side = Side('thin')
# 定义表头边框样式,有底边和右边
header_border = Border(bottom=side, right=side)
# 定义表中、表尾边框样式,有左边
content_border = Border(left=side)
# 设置文件夹路径
path = './各部门利润表汇总_副本/'
# 返回当前目录下所有文件名
files = os.listdir(path)
# 循环文件名列表
for file in files:
# 拼接文件路径
file_path = path + file
# 打开工作簿
wb = load_workbook(file_path)
# 打开工作表
ws = wb.active
# 调整列宽
ws.column_dimensions['A'].width = 10
ws.column_dimensions['B'].width = 25
ws.column_dimensions['C'].width = 50
ws.column_dimensions['D'].width = 10
ws.column_dimensions['E'].width = 20
ws.column_dimensions['F'].width = 15
# 循环第一行单元格,调整表头样式
for cell in ws[1]:
# 设置单元格填充颜色
cell.fill = header_fill
# 设置单元格对齐方式
cell.alignment = align
# 设置单元格边框
cell.border = header_border
# 获取最后一行行号
row_num = ws.max_row
# 从第二行开始,循环到倒数第二行
for row in ws.iter_rows(min_row=2, max_row=(row_num-1)):
# 循环取出单元格,调整表中样式
for cell in row:
cell.fill = content_fill
cell.alignment = align
cell.border = content_border
# 循环最后一行单元格,调整表尾样式
for cell in ws[row_num]:
cell.fill = bottom_fill
cell.alignment = align
cell.border = content_border
# 保存
wb.save(file_path)
案例8 绘制Excel图表
import os
from openpyxl import load_workbook
from openpyxl.chart import LineChart, Reference
# 设置目标文件夹路径
path = './部门利润表汇总/'
# 获取文件夹下的所有文件名
file_list = os.listdir(path)
# 遍历文件名列表,取得每一个文件名
for file_name in file_list:
# 拼接文件路径
file_path = path + file_name
print('正在处理:' + file_name)
# 读取工作簿
wb = load_workbook(file_path)
# 读取工作簿中的活跃工作表
ws = wb.active
# 实例化 BarChart 类,得到 BarChart 对象 条形图
#chart = BarChart()
# 实例化 LineChart 类,得到 LineChart 对象 折线图
chart = LineChart()
# 引用工作表的部分数据
data = Reference(worksheet=ws, min_row=3, max_row=9, min_col=1, max_col=5)
# 添加被引用的数据到 LineChart 对象
chart.add_data(data, from_rows=True, titles_from_data=True)
# 添加 LineChart 对象到工作表中,指定折线图的位置
ws.add_chart(chart, "C12")
# 引用工作表的表头数据
cats = Reference(worksheet=ws, min_row=2, max_row=2, min_col=2, max_col=5)
# 设置类别轴的标签
chart.set_categories(cats)
# 设置 x 轴的标题
chart.x_axis.title = "季度"
# 设置 y 轴的标题
chart.y_axis.title = "利润"
# 改变线条颜色
chart.style = 48
# 保存文件
wb.save(file_path)
# 在终端提示表格绘图结束
print('恭喜你,工作表中的图绘制成功!')