python实现词云图

python实现词云图

import matplotlib.pyplot as plt
import jieba
from wordcloud import WordCloud,ImageColorGenerator
from scipy.misc import imread
text = open(r'.\XXX讲话讲话.txt','r').read() #导入文本
#print(text)
cut_text = jieba.cut(text)
#print(type(cut_text))
#连接词条
result = ' '.join(cut_text)
#print(result)
#初始化自定义背景
image = imread(r'.\timg.jpg')

#生成词云图
wc = WordCloud(
#字体路径
font_path='C:\Windows\Fonts\simkai.TTF',
#背景颜色
background_color='white',
#width=500,
#height=350,
#图片最大字体
max_font_size=50,
#图片最小字体
min_font_size=10,
mask = image
)
#插卡
wc.generate(result)
#从背景里面提取背景颜色
image_color = ImageColorGenerator(image)
wc.recolor(color_func=image_color)
wc.to_file(r'.\wordcloud.png')

#显示图片
#指定图片的名称
plt.figure('jay')
plt.imshow(wc)
#去掉坐标轴
plt.axis('off')
plt.show()