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)

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