发布时间:2023-02-01 14:00
先安装所需要的库
from wordcloud import WordCloud
import jieba
import matplotlib.pyplot as plt
#绘制词云
def draw_wordcloud():
comment_text = open('kouhong.txt','r').read()
cut_text = " ".join(jieba.cut(comment_text)) #结巴分词
color_mask=plt.imread("white.png") #读取背景图片
cloud = WordCloud(
font_path='simsun.ttc', #设置字体
background_color='white', #设置背景色
mask=color_mask, #词云形状
max_words=300,#允许最大词汇
max_font_size=80,#最大号字体
collocations=False#避免重复词
)
word_cloud = cloud.generate(cut_text) #产生词云
#word_cloud.to_file("kouhong.jpg") #保存图片
plt.imshow(word_cloud) #显示词云图片
plt.axis('off')
plt.show()
if __name__ == '__main__':
draw_wordcloud()