Python入门（三）—— 文字转图片处理

前言

继上一篇文章的需求，爬取新闻早报，拼成一张图片。本文说明如何将新闻文字排版成一张图片。

准备环境

引入图片编辑库

我们用PIL(Python Imaging Library)这个库来实现P图，执行pip install pillow脚本安装（PIL仅支持到Python 2.7，在Python 3.x的版本中我们可以使用他的开源版Pillow）。安装执行如图：

编码实现

1. 实现简单的程序画图

代码如下 (代码#号后面的内容表示注释，便于理解，不会影响程序运行)：


from PIL import Image, ImageDraw, ImageFont  # 引入图片，画图笔，图片字体三个库

img_1 = Image.new('RGB', (750, 2000), (255, 255, 255)) # 建一张新图，颜色用RGB，尺寸 750x2000，底色三个255表示纯白
draw = ImageDraw.Draw(img_1) # 创建一个画笔

header_position = (130,200) # 标题的横纵坐标位置
header_font = ImageFont.truetype('C:/Windows/Fonts/simkai.ttf', 55) # 标题的字体，字号55
draw.text(header_position , '互联网日报', '#726053', header_font) # 参数分别是坐标，文字内容，文字色号，文字字体

img_1.show() # 弹框展示图片

执行效果：

会弹出一张图片，上面用楷书写着’互联网日报‘

2. 预加工新闻文本

之前获取到的新闻内容，有几个小问题：

1. 内容中有一些没用的标签，例如 <p><br/></p>我们要去掉 2. 有一些空白的新闻，我们要过滤掉例如 3、<br/>

用以下代码进行去冗余工作：

news_content = str(news_content) # 将html标签转成纯字符串
news_content = news_content.replace('<p>', "") # <p>标签替换成空字符串
news_content = news_content.replace('</p>', "") # </p>标签替换成空字符串
news_content = news_content.replace('<br/>', "\n")  # '<br/>'标签替换成系统换行符
news_content = news_content.splitlines()  # 按行分割，返回结果是一个数组，数组元素是单行的文字

最后，我们还需要将文本分割成每行限定25字的句子，否则句子长度不同会导致文字溢出图片。相关代码


import textwrap # 该库用于手动换行文字
news_wrap = [] # 准备一个数组来装我们的结果集
for line in news_content: # 循环遍历数组中的每行文字，line是临时变量，指代当前所循环到的文字
    if len(line) < 4:  # 左边有四个空格， len(line)表示计算该行的字数，小于4个字的就舍弃
        continue   # 左边有八个空格，
    elif len(line) < 25:   #  若字数大于4个且小于25个 （限制每行字数不超过25字）
        news_wrap.append(line)   # 添加到数组中
    else:    # 若字数大于25个字
        wrap = textwrap.wrap(line, 25) # 按每行25个字分割成数组
        news_wrap = news_wrap + wrap  # 拼到结果数组中
 print(news_wrap) # 分割后的数组打印出来看看

用到的函数说明

💡

1. str(xxx) ，这个函数可以将其他数据格式，转换成纯字符串，便于进行字符串的查找替换处理 2. xxx.replace('a','b')，这个函数在字符串中可以用，将字符串中的 a 替换成 b; 3. xxx.splitlines()，这个函数将整段文字，按照换行符，分割成一组字符： 4. for xx in xxx: 遍历循环，每次从xxx中取出一个元素进行处理 5. if xxx: 逻辑判断条件，满足该条件则进入冒号后紧跟着的代码块 6. elif xxx: 如果不满足上满的条件，但是满足这个条件，则进入 7. else: 如果都不满足，则进入 8. len(xx) 计算xx的长度，如果xx是一个字符串，则计算字符串的字数;len = lenght的缩写。9. textwrap.wrap(xx,25); 将这个字符串分割成每行不超过25个字的数组。

经过处理后的文字效果：

3. 用代码在图片上绘制

创建图片并写入文字的程序步骤思路：

1. 引入图片、画笔、字体 2. 根据新闻行数，生成一个空白图片 3. 在图片上创建一个画笔 4. 指定写入文字的位置、字体、内容、颜色，然后从上往下逐行写入 5. 用一个变量记录当前写到图片的位置高度，每次画笔写入后，画笔的位置向下方移动特定的高度 6. 完成后展示、或保存图片

代码演示:

from PIL import Image, ImageDraw, ImageFont  # 引入图片，画图笔，图片字体三个库
IMG_SIZE = (900, len(news_wrap) * 44) # 图片尺寸 900x新闻行数x每行行高
img_1 = Image.new('RGB',IMG_SIZE  , (255, 255, 255))  # 建一张新图，颜色用RGB，，底色三个255表示纯白
draw = ImageDraw.Draw(img_1)  # 创建一个画笔

header_position = (60, 30)  # 标题的横纵坐标位置
header_font = ImageFont.truetype('simkai.ttf', 55)  # 标题的字体楷体，字号55
draw.multiline_text(header_position, '互联网日报', '#726053', header_font)  # 入参分别是坐标，文字内容，文字色号，文字字体

current_height = 100
for line in news_wrap:
    if line.startswith('【'):
        news_font = ImageFont.truetype('simkai.ttf', 45)  # 标题的字体楷体，字号50
        draw.text((60, current_height + 30), line, '#726053', news_font)
        current_height += 80
    else:
        news_font = ImageFont.truetype('simkai.ttf', 30)  # 新闻字体30
        draw.text((60, current_height), line, '#726053', news_font)
        current_height += 40

img_1.show()  # 弹框展示图片
img_1.save('早报.jpg') # 保存成文件

执行效果：

如果图片会直接弹框展示，恭喜你已经完成了这部分的内容；字体、颜色排班可以根据自己的想法调整试试~

总结

有了PIL这个库，python可以很方便地进行绘图，这篇文章的难点在于要汇入图片的文本内容清洗、以及图片元素在图片上的排版尺寸与位置。