wordcloud词云模块

wordcloud词云模块

下载

pip install wordcloud

使用

import wordcloud##调用整个模块

form wordcloud import WordCloud##调用wordcloud中的WordCloud子模块

1.配置模块参数

2.加载文本.generate()

3.输出词云文件.to_file(" ")

from wordcloud import WordCloud
from imageio import imread
import matplotlib.pyplot as plt
import jieba
f = open('中国十九大.txt', encoding="utf-8")#打开文件并转码
date = f.read()#读取文件内容
res = jieba.lcut(date)#将文件内容切成字符串列表
# print(res)
result = "".join(res)#将切成字符串列表后的文件转成文本
# print(result)
f.close()
mask_1 = imread("五角星.jpg")#引入图片模型（面具）使用imageio模块\
#或者使用scipy模块from scipy.misc import imread
wc = WordCloud(background_color="red",
               font_path=r'C:\Windows\Fonts\STFANGSO.TTF',
               width=1960,
               height=1080,
               mask=mask_1)#设置面具
#定义WordCloud参数
wc.generate(result)#向WordCloud中的wc加入转换后的文本
wc.to_file('练习.png')#将转换后的文本存为词云文件
plt.imshow(wc)
plt.show()