漂亮的词云图是怎么生成的？不到10行代码搞定！ - 文章 - 开发者社区

猴哥的第 40 期分享，欢迎追看

后台很多小伙伴对词云感兴趣：给一段文本，然后根据其中词语出现的频率，生成好看的词云，像下面这张图一样：

picture.image

生成这个其实很简单，几行 Python 代码就能搞定，今天就来带大家实操一番。

环境准备

配置好 Python 环境后，需要安装两个包：

jieba：用于分词
wordcloud：用于生成词云

  
pip install jieba  
pip install wordcloud

获取模板图片（可选）

wordcloud 中默认生成的是矩形图片。

如果希望生成的词云图片具有特定的样式，你需要准备一张 png 格式的含有透明图层的图片，像下面这样：

picture.image

怎么获取 png 格式的图片？

打开：https://www.remove.bg/zh/upload

上传一张图片，然后点击下载即可：

picture.image

获取字体文件

Windows 电脑中，字体默认保存在 C:\Windows\Fonts\，文件后缀为 .ttf。

找到想要想要生成的字体路径。

获取文本信息

大家可以试试自己的微信聊天记录。

这里我们以《红楼梦》小说为例进行演示，输入的是 .txt 文本文件。

  
import jieba  
def cut_words(text):  
    # 使用 jieba 分词  
    words = jieba.cut(text)  
    return ' '.join(words)  
with open('D:\Downloads\data\红楼梦.txt', 'r', encoding='utf-8') as f:  
        text = f.read()  
text = cut_words(text)

生成词云图

最后，初始化一个 wordcloud 实例，把刚刚分词后的文本输入进来，生成最终的词云图片。

  
import wordcloud  
def generate_wordcloud(text, stopwords=None, mask=None, max_words=50, img_name='1.jpg'):  
    wordcloud = WordCloud(width=800, height=400,   
                        mask=mask,  
                        background_color='white',  
                        stopwords=stopwords,  
                        font_path='C:\Windows\Fonts\simkai.ttf', # simkai.ttf STXINGKA.TTF  
                        max_words=max_words,  
                        ).generate(text)  
    wordcloud.to_file(img_name)