Python绘制PDF文件~超简单的小程序

Python绘制PDF文件

项目简介
这次项目很简单，本次项目课，代码不超过40行，主要是使用 urllib和reportlab模块，来生成一个pdf文件。

reportlab官方文档

http://www.reportlab.com/docs/reportlab-userguide.pdf
我们看看这个网页上的原数据：
http://www.swpc.noaa.gov/ftpdir/weekly/Predict.txt


代码：

#-*- coding: utf-8 -*-

# 1. 用于下载原数据
import urllib

# 2. 使用shapes可以画出复杂的形状
from reportlab.graphics.shapes import *

# 3. chars包里包含许多常用的图形
from reportlab.graphics.charts.lineplots import LinePlot
from reportlab.graphics.charts.textlabels import Label

# 4. 用于渲染PDF文件
from reportlab.graphics import renderPDF

URL = 'http://www.swpc.noaa.gov/ftpdir/weekly/Predict.txt'
COMMENT_CHARS = '#:'

# 5. 初始化坐标原点
drawing = Drawing(400, 200)
data = [ ] # 用于存储下载下来并过滤好的数据

# 6. 下载数据并提取有效数据
for line in urlopen(URL).readlines():
    if not line.isspace() and not line[0] in COMMENT_CHARS:
        data.append([float(n) for n in line.split()])

# 7. 提取用于画图的数据
pred = [row[2] for row in data]
high = [row[3] for row in data]
low = [row[4] for row in data]
times = [row[0] + row[1]/12.0 for row in data]

lp = LinePlot()
lp.x = 50
lp.y = 50
lp.height = 125
lp.width = 300

# 8. zip()是Python的一个内建函数，它接受一系列可迭代的对象作为参数，将对象中对应的元素打包成一个个tuple（元组），然后返回由这些tuples组成的list（列表）。若传入参数的长度不等，则返回list的长度和参数中长度最短的对象相同。
lp.data = [zip(times, pred), zip(times, high), zip(times, low)]

lp.lines[0].strokeColor = colors.blue
lp.lines[1].strokeColor = colors.red
lp.lines[2].strokeColor = colors.green

drawing.add(lp)
drawing.add(String(250, 150, 'Sunspots', fontSize=14, fillColor=colors.red))

renderPDF.drawToFile(drawing, 'report3.pdf', 'Sunspots')

说明：
注释6：line.isspace()判断改行是不是为空行注释7：data这个列表的每个元素实际上也是列表，筛选数据的依据如下图：

其实需要的就是年(times=YR)，预测数据(pred=PREDICTED)，最高数据(high=HIGH)，最低数据(low=LOW)
注释8：因为画图需要用的数据，必须是一定格式的，所以用zip组装了数据:[[ (,), (,) ..... ]]



程序运行结果：



今天PO了一个小程序，非常简单，大家就当练练手好了~最近小编很苦恼不知道 继续发些什么，如果有什么建议可以告诉小编那~啦啦啦~

更多基础课、项目课欢迎登陆实验楼官方网站：http://www.shiyanlou.com​