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如下:
生成的词云如下: