Python使用wordcloud+pillow基于给定图像制作词云

大致思路：Python扩展库wordcloud可以用来制作词云，而pillow库提供了图像处理功能，代码结合二者创建了词云头像，其中把给定的图像作为参考，只保留词云中与图像前景对应位置的像素，起到裁剪作用。

import random
import string
import wordcloud
from PIL import Image

def show(s):
    # 创建wordcloud对象
    wc = wordcloud.WordCloud(
        r'C:\windows\fonts\simfang.ttf',
        width=500, height=400,
        background_color='white',
        font_step=3,
        random_state=False,
        prefer_horizontal=0.9)
    # 创建并显示词云
    t = wc.generate(s)
    t.to_image().save('t.png')

# 如果空间足够，就全部显示
# 如果词太多，就按频率显示，频率越高的词越大
show('hello world 董付国 董付国 董付国 董付国  abc fgh yhnbgfd 董付国 董付国 董付国 董付国 Python great Python Python')

上面代码某次运行生成的图片为：

def create(imgFile, s):
    im = Image.open(imgFile)
    w, h = im.size
    # 创建wordcloud对象
    wc = wordcloud.WordCloud(
        r'C:\windows\fonts\simfang.ttf',
        width=w, height=h,
        background_color='white',
        font_step=3,
        random_state=False,
        prefer_horizontal=0.9)
    # 创建并显示词云
    t = wc.generate(s)
    t = t.to_image()
    for w1 in range(w):
        for h1 in range(h):
            if im.getpixel((w1,h1))[:3] == (255,255,255):
                t.putpixel((w1,h1), (255,255,255))

    t.save('result.png')

chs = string.ascii_letters + string.digits + string.punctuation

s = [''.join((random.choice(chs)for i in range(8))) for j in range(650)]
s = ' '.join(s)
create('test1.png', s)

上面代码测试图像和生成的词云图像分别为

更多Python技术文章请关注微信公众号“Python小屋”进行阅读