用Python制作数据报告:如何自动生成PDF格式的报告?
最近在琢磨数据分析工作的自动化,手动做报告真是太费劲啦!试过用Python整了个自动生成PDF报告的小工具,效果还不错。今天就聊聊怎么用Python把数据处理、可视化和PDF生成一条龙搞定。
reportlab:PDF生成的得力助手
reportlab是Python里特别好用的PDF生成库,搭配起来做报表简直不要太方便。装它超简单:
pip install reportlab
来个最基础的PDF示例:
from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import A4
# 新建个PDF文件
pdf = canvas.Canvas(“简单报告.pdf”, pagesize=A4)
pdf.drawString(100, 750, “这是我的第一份Python生成的报告”)
pdf.save()
温馨提示:记得指定pagesize,不然默认页面可能有点小
数据可视化:让报告更生动
光有文字多没意思,加点图表才够味儿。matplotlib这个库画图超好用:
import matplotlib.pyplot as plt
import numpy as np
# 造点数据
sales = [120, 150, 180, 210, 165]
months = ['1月', '2月', '3月', '4月', '5月']
# 画个柱状图
plt.figure(figsize=(8, 6))
plt.bar(months, sales)
plt.title('销售数据统计')
plt.savefig('sales_chart.png')
组合拳:图文并茂的报告
把文字和图片都塞进PDF里:
from reportlab.lib.utils import ImageReader
def create_report():
pdf = canvas.Canvas(“完整报告.pdf”, pagesize=A4)
# 加标题
pdf.setFont(“Helvetica-Bold”, 16)
pdf.drawString(100, 750, “2024年第一季度销售报告”)
# 放图片
img = ImageReader('sales_chart.png')
pdf.drawImage(img, 50, 400, width=500, height=300)
# 加说明文字
pdf.setFont(“Helvetica”, 12)
pdf.drawString(100, 350, “从图表可以看出,销售额整体呈上升趋势”)
pdf.save()
进阶玩法:自动处理表格
要是数据多,可以用platypus模块处理表格:
from reportlab.platypus import SimpleDocTemplate, Table, TableStyle
from reportlab.lib import colors
def create_table_report():
doc = SimpleDocTemplate(“表格报告.pdf”, pagesize=A4)
# 准备表格数据
data = [
['月份', '销售额', '同比增长'],
['1月', '120万', '15%'],
['2月', '150万', '25%'],
['3月', '180万', '20%']
]
# 设置表格样式
table = Table(data)
table.setStyle([
('GRID', (0, 0), (-1, -1), 1, colors.black),
('BACKGROUND', (0, 0), (-1, 0), colors.grey)
])
doc.build([table])
温馨提示:表格样式可以自己调整,比如改颜色、字体大小啥的
完整代码示例:
from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import A4
from reportlab.lib.utils import ImageReader
from reportlab.platypus import SimpleDocTemplate, Table, TableStyle
from reportlab.lib import colors
import matplotlib.pyplot as plt
import numpy as np
def generate_chart():
sales = [120, 150, 180, 210, 165]
months = ['1月', '2月', '3月', '4月', '5月']
plt.figure(figsize=(8, 6))
plt.bar(months, sales)
plt.title('销售数据统计')
plt.savefig('sales_chart.png')
plt.close()
def create_full_report():
# 先生成图表
generate_chart()
# 创建PDF
pdf = canvas.Canvas(“数据分析报告.pdf”, pagesize=A4)
# 添加内容
pdf.setFont(“Helvetica-Bold”, 16)
pdf.drawString(100, 750, “2024年第一季度销售报告”)
img = ImageReader('sales_chart.png')
pdf.drawImage(img, 50, 400, width=500, height=300)
pdf.setFont(“Helvetica”, 12)
pdf.drawString(100, 350, “销售情况分析说明:”)
pdf.drawString(120, 330, “1. 销售额持续增长”)
pdf.drawString(120, 310, “2. 第四月份达到峰值”)
pdf.save()
# 运行试试看
create_full_report()
搞定啦!代码运行完就能看到生成的PDF报告。要是想改报告样式,调整一下坐标和格式就好。试着改改参数,做出自己想要的报告样式吧!