Python文本挖掘：词云图（学习笔记）

先安装所需要的库

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()

结果：