python WordCloud制作词云

1. 代码

from wordcloud import WordCloud
import matplotlib.pyplot as plt
import numpy as np
from PIL import Image

#打开文本

text = open(‘E:/python_test/test.txt’,‘r’, encoding=‘UTF-8’).read()

#simhei.ttf是从C:\Windows\Fonts文件夹下直接复制过来的字体文件

wc = WordCloud(font_path=‘E:/python_test/simhei.ttf’,
mode=‘RGBA’, background_color=None).generate(text)

#显示词云

plt.imshow(wc, interpolation=‘bilinear’)
plt.axis(‘off’)
plt.show()

其中test.txt的内容如下截图：

2.加蒙版mask

这样就可以跟进选择的图片生成各种各样的词云了，样子和mask一样。

text = open(‘E:/python_test/test.txt’,‘r’, encoding=‘UTF-8’).read()
mask = np.array(Image.open(“E:/python_test/heart.jpg”))

wc = WordCloud(font_path=‘E:/python_test/simhei.ttf’, mask=mask,
margin=1, random_state=1, background_color=‘white’).generate(text)

plt.imshow(wc, interpolation=‘bilinear’)
plt.axis(‘off’)
plt.show()

mask如下：

生成的词云如下：