Python学习小组课程P5-Python办公（2）Excel读取与Word生成

一、前言

注意：此为内部小组学习资料，非售卖品，仅供学习参考。
本系列课程：
Python学习小组课程-课程大纲与Python开发环境安装
Python学习小组课程P1-Python基础（1）语法与数组
Python学习小组课程P2-Python基础（2）文件操作
Python学习小组课程P3-Python爬虫（1）HTML与Json解析
Python学习小组课程P4-Python办公（1）Excel保存
Python学习小组课程P5-Python办公（2）Excel读取与Word生成
Python学习小组课程P6-Python办公（3）邮件与钉钉消息通知

二、知识点

1 读取Excel文件，分析并展示数据

知识点：Excel读取操作

更多使用方法可参考《python 使用 openpyxl 修改表格中的内容》

import openpyxl
import matplotlib.pyplot as plt

# 支持中文
plt.rcParams['font.sans-serif'] = ['SimHei']  # 用来正常显示中文标签
plt.rcParams['axes.unicode_minus'] = False  # 用来正常显示负号

path='P5.1_data/销售数据100019645157.xlsx'
wb=openpyxl.load_workbook(path) #创建工作簿
sheet=wb.active #获取活动表

rows=sheet.max_row
cols=sheet.max_column


size_dic={}
for row in range(1,rows+1):
    size=sheet.cell(row,3).value

    if size not in size_dic:
        size_dic[size]=1
    else:
        size_dic[size]+=1

#sheet.cell(1,cols+1).value='新增数据'
#wb.save(path)

print(size_dic)


size_list=[]
for size in size_dic:
    size_list.append([size,size_dic[size],size_dic[size]/10.])

labels=[item[0] for item in size_list]
fraces=[item[2] for item in size_list]

print(size_list)

plt.pie(x=fraces,labels=labels,autopct='%1.1f%%')
plt.show()

{'8(男41)': 3, '8.5(男42)': 2, '8.5(新42)': 1, '9(男42.5)': 1, '7(男40)': 2, '9.5(男43)': 1}
[['8(男41)', 3, 0.3], ['8.5(男42)', 2, 0.2], ['8.5(新42)', 1, 0.1], ['9(男42.5)', 1, 0.1], ['7(男40)', 2, 0.2], ['9.5(男43)', 1, 0.1]]

2 生成Word文档

知识点：Word文档生成

安装python-docx库

pip install python-docx

from docx import Document
from docx.shared import RGBColor 
import os

doc=Document()
doc.add_heading('如何使用Python创建Word文档？',0)

doc.add_heading('如何使用Python创建Word文档？',1)
doc.add_heading('如何使用Python创建Word文档？',2)
doc.add_heading('如何使用Python创建Word文档？',3)

p1=doc.add_paragraph('这是段落1\n')

#添加样式
run1=p1.add_run('这是内容1.1\n')
run1.bold=True

p1.add_run('颜色改变').font.color.rgb=RGBColor(255,0,0)

#创建目录
path='P5.2_data/'
if not os.path.exists(path):
    os.makedirs(path)

doc.save(f'{path}word文档创建.doc')

3 根据模板生成Word文档

知识点：Word文档读取、生成

from docx import Document
import os

infos=[
    ['00001',2030,12,12,12,0,'闯红灯',500],
    ['00002',2031,12,12,12,0,'违反禁令',300],
    ['00003',2032,12,12,12,6,'违章停车',300],
]

for info in infos:
    doc=Document('P5.3_data/word模板.docx')
    for p in doc.paragraphs:
        for run in p.runs:
            run.text=run.text.replace('{0}',info[0])
            run.text=run.text.replace('{1}',str(info[1]))
            run.text=run.text.replace('{2}',str(info[2]))
            run.text=run.text.replace('{3}',str(info[3]))
            run.text=run.text.replace('{4}',str(info[4]))
            run.text=run.text.replace('{5}',str(info[5]))
            run.text=run.text.replace('{6}',info[6])
            run.text=run.text.replace('{7}',str(info[7]))
    
    doc.save(f'P5.3_data/车辆{info[0]}.docx')