根据图片形状设置词云

from wordcloud import WordCloud
import numpy as np
from PIL import Image

#步骤一 加载数据文件
file = open('data/alice.txt',mode='r')
text = file.read()
#步骤二 加载背景图数据
mask = np.array(Image.open('data/alice_mask.png'))
#步骤三 建立词云对象(默认黑色)
wd = WordCloud(max_font_size=40,mask=mask,background_color='white')
wd.generate(text)
#步骤四 生成图像
wd.to_file('alice_wd.png')
posted @ 2018-12-28 16:20  淘气小菲菲  阅读(1224)  评论(0编辑  收藏  举报