Python 畅玩wordcloud词云!


Python的wrodcloud模块生成词云只需要二十几行代码就便可以轻松实现!效果图1是读取《新世纪福音战士》剧情解析的词云。
软件平台为Pycharm(社区版本)
效果图1的代码:
from wordcloud import WordCloud
from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
import jieba #结巴模块是用于对文本内容中的文字进行解词
# 注意!使用的图片背景必须是纯白色,mask才生效。
mask=np.array(Image.open('xx-3.png'))
#设置背景色,词组重复;词组最多出现200次,画布宽度,画布高度
#遮罩的轮廓线宽度, 遮罩的轮廓线颜色
wcd=WordCloud(background_color='white',repeat=True ,font_path='front/msyh.ttc',mask=mask,max_words=200,width=500,height=500,contour_width=1,contour_color='blue')
#结巴分词
#读取eva_back.txt文本
test=open('eva_back.txt','r',encoding='UTF-8')
tt=test.read() #读取test的内容
ss=' '.join(jieba.lcut(tt))
wcd.generate(ss)
wcd.to_image()
wcd.to_file('w-test-01.jpg') #保存生成的图片
#显示图像
plt.imshow(wcd)
plt.axis("off") #关闭绘制图像坐标轴
plt.show()