Advertisement

python可视化词云

阅读量:

完整代码

复制代码
    # -*- coding: utf-8 -*-
    from os import path
    import matplotlib.pyplot as plt
    from scipy.misc import imread
    import jieba
    from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator
    
    d = path.dirname('.')
    # 读取文本
    text = open(path.join(d, 'E:/pythonproject/ali.txt')).read()
    # 结巴中文分词,生成字符串
    cut_text = jieba.cut(text)
    result = " ".join(cut_text)
    
    # 设置背景图片
    JPG = imread(path.join(d, "E:/pythonproject/ali.jpg"))
    # 设置参数:背景颜色,词云显示的最大词数,设置背景图片,字体最大值
    wc = WordCloud(
    # 设置字体,不指定就会出现乱码
    font_path=r'.\simhei.ttf',
    background_color="white",
    max_words=2000,
    mask=JPG,
    stopwords=STOPWORDS.add("said"),
    max_font_size=50,
    random_state=10)
    # 产生词云
    wc.generate(result)
    
    # 从背景图片生成颜色值
    image_colors = ImageColorGenerator(JPG)
    #保存图片
    wc.to_file(path.join(d, 'E:/pythonproject/result.jpg'))
    
    plt.imshow(wc)
    plt.axis("off")
    plt.show()

原图

这里写图片描述

效果图

这里写图片描述

全部评论 (0)

还没有任何评论哟~