利用Python及其第三方库绘制词云图

深藏阁楼爱情的钟 2023-02-18 08:17 212阅读 0赞

你一定在各种网站（bushi）见过这种图：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ1NjkwMDUw_size_16_color_FFFFFF_t_70_pic_center]  
这种图就是词云图了

当然，词云的意义并不仅仅局限于欣赏，更多的是应用于数据可视化分析。

由于网上好多词云生成器不是崩了就是付费…于是我下定决心自学词云（瞬间感觉自己好卑微啊23333）。利用Python和一些第三方库，如本次词云生成使用了jieba、wordcloud、imageio这三个库。

**同样的，本次仍然不介绍这三个库，下次一定（咕咕咕）**

废话不说，直接上代码：

#!/usr/bin/env Python
    # coding=utf-8
    
    #  @author:神乐坂几禾
    #  @time:2020-6-2 01:26
    #  @discribe:wordcloud of "Protection of Endangered Animals"
    
    from wordcloud import WordCloud,ImageColorGenerator
    import jieba
    import imageio
    
    # 词云背景模板读取
    mk2=imageio.imread("bird.png")
    
    # 外部文本读取
    t=open('text.txt','r').read()
    
    # jieba分词
    list=jieba.lcut(t)
    string=" ".join(list)
    
    # 构建词云对象w
    w=WordCloud(background_color='black',
                mask=mk2,
                font_path='msyh.ttc',
                scale=40,
                height=800)
    w.generate(string)
    
    # 提取模板颜色
    colors=ImageColorGenerator(mk2)
    
    # 新建词云对象，重新上色
    w_withColor=w.recolor(color_func=colors)
    w_withColor.to_file('post.png')

自己写的程序是很灵活的，可以根据自己的喜好来修改生成词云图的尺寸、背景颜色、清晰度、形状，乃至字体。

这里需要注意以下几点：

1.  由于**imageio只能识别背景为白色的图片**，所以如果背景不是白色的话，可能你还需要用ps扣下图。  
    如果你的背景图片不是白色的，出来的词云图形状会仍然是矩形；
2.  文本文件最好和程序在同一项目目录下；
3.  可能受到一些影响的是，该程序会对文本去重（好狗血），所以编写文本的时候就会惨不忍睹（如果直接找的文本当我没说）；
4.  项目名称不能是wordcloud，否则会报错（以后会 单独出博客来讲解）。

最终我也是做了一张图，成品太大了，截个图吧：  
![在这里插入图片描述][20200616172835113.png_pic_center]  
果然自己写还是省钱的呀23333

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ1NjkwMDUw_size_16_color_FFFFFF_t_70_pic_center]: https://img-blog.csdnimg.cn/20200616171357317.jpg?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ1NjkwMDUw,size_16,color_FFFFFF,t_70#pic_center
[20200616172835113.png_pic_center]: https://img-blog.csdnimg.cn/20200616172835113.png#pic_center