本文中的所有示例代码及素材均在 github 项目「shijiu_wordcloud」的 wordcloud1 文件夹中。

python词云模块的安装

制作词云的主要模块为 wordcloud，另外，要使用 jieba 库切割中文句子。在命令行中执行以下命令即可安装：

pip install wordcloud jieba

制作你的第一个词云

下面用四行代码制作了一个词云，代码中包含了详细的注释：

# example1.py
# 从词云模块 wordcloud 导入 WordCloud 类
from wordcloud import WordCloud

# 创建一个 WordCloud 对象 wc
wc = WordCloud()

# 将字符串传入 wc 对象的 generate 方法（此时只能处理英文）
wc.generate('Today you do things people will not do, tomorrow you will do things people can not do.')

# 利用 to_file 方法将词云输出到指定路径
wc.to_file(r'词云输出\first_wordcloud.png')

first_wordcloud.png

改变背景颜色，不再“黑乎乎”

上面制作的词云的显得黑乎乎的，不好看，此时需要更改词云的背景颜色。

只要简单改变上述创建 WordCloud 对象的代码，即可得到不同背景颜色的词云：

# example2.py
# 设置背景颜色为白色
wc = WordCloud(background_color='white')

white_background.png

python还支持很多颜色，下表列出了一部分，有兴趣的读者可自行尝试：

available_color

制作中文词云

可能有的读者已经发现，上面的代码使用中文字符串时，输出的词云会显示一堆大小不一的方框，看不到中文的影子。

原因在于 WordCloud 使用的默认字体没有为中文编码，所以没办法显示中文。

要显示中文同样很简单，只需在创建 WordCloud 对象时通过设置 font_path 参数设置支持中文的字体即可：

# example3.py
# 通过 font_path 参数设置支持中文的字体
wc = WordCloud(background_color='white', font_path='SIMLI.TTF')

其中，字体的名称 SIMLI.TTF 表示隶书，如果读者电脑有这种字体，就能正常显示中文了。

chinese.png

另外，Windows 系统可以在 C:\Windows\Fonts 路径下找到能使用的字体，通过 查看 -> 选择详细信息 -> 勾选字体文件名称 的操作，显示字体文件名称，然后就可以如图所示地找到字体对应的英文名称了。

查看字体名称

把中文句子“切开”

上面制作的中文词云，中文是整句整句显示的，不是我们想要的一个一个的词。

要制作由一个一个词组成的词云，就要借助 jieba 库把句子切割成词：

# example4.py
# 从 jieba 模块导入切割句子的函数 lcut
from jieba import lcut

text = '永远掌握在自己手中'

# 将句子切开
text = lcut(text)
print(text)

运行后得到如下结果：

['永远', '掌握', '在', '自己', '手中']

中文分词词云

话不多说，先上代码：

# example5
from wordcloud import WordCloud
from jieba import lcut

text = '有的人循规蹈矩，一生碌碌无为，到老之时唉声叹气，责怪命运的不公平；' \
       '而有的人，善于创新，关键时放手一搏，即使生命短暂，也活出了不一样的精彩，不一样的闪耀。' \
       '命运，永远掌握在自己手中，但要看你是怎样对待它的。'
text = lcut(text)

# 利用空格将分散的中文词练成一个单独的字符串，注意引号内有一个空格
text = ' '.join(text)

wc = WordCloud(background_color='white', font_path='SIMLI.TTF')
wc.generate(text)
wc.to_file('词云输出\\partition.png')

partition.png

值得注意的是，利用 jieba 切割句子后，得到的是多个词组成的列表，但 WordCloud 的 generate 方法要求传入一个单独的字符串，所以要把单词用空格连成一个字符串。

连词成串用到 join 方法，注意上面代码的引号中有一个空格。

一词成云：repeat 参数

用一个单词做词云，只需在创建 WordCloud 对象时设置 repeat 参数为 True：

# example6.py
from wordcloud import WordCloud

# 设置 repeat 参数为 True
wc = WordCloud(background_color='white', repeat=True)

wc.generate('star')
wc.to_file('词云输出\\repeat.png')

repeat.png

单一颜色的词云

制作单一颜色的词云，需要在创建 WordCloud 对象时设置 color_func 函数。读者只需更改下面代码的颜色即可体验其他颜色：

# example7.py
from wordcloud import WordCloud

# 设置 color_func 函数， 读者只需修改最后的颜色即可
wc = WordCloud(background_color='white', repeat=True, color_func=lambda *args, **kwargs: "deepskyblue")

wc.generate('star')
wc.to_file('词云输出\\singlecolor.png')

singlecolor.png

制作高清词云：scale 参数

细心的朋友可能已经发现，前面制作的词云清晰度普遍较低，如何制作高清词云呢？

其实只需将 scale 参数设置成更大的值（默认值为 1）：

# example8.py
from wordcloud import WordCloud

# 将 scale 参数设置为 10
wc = WordCloud(background_color='white', repeat=True, scale=10)

wc.generate('star')
wc.to_file('词云输出\\highdefinition.png')

highdefinition.png

设置“停用词”

停用词用于指定词云不显示哪些词，在代码中将 stopwords={'词1', '词2', ...} 传入 WordCloud 类即可：

# example9.py
from wordcloud import WordCloud
from jieba import lcut

text = '有的人循规蹈矩，一生碌碌无为，到老之时唉声叹气，责怪命运的不公平；' \
       '而有的人，善于创新，关键时放手一搏，即使生命短暂，也活出了不一样的精彩，不一样的闪耀。' \
       '命运，永远掌握在自己手中，但要看你是怎样对待它的。'
text = lcut(text)
text = ' '.join(text)

# stopwords 的词用 大括号 括起来
wc = WordCloud(background_color='white', font_path='SIMLI.TTF', stopwords={'一样', '老之时'})

wc.generate(text)
wc.to_file('词云输出\\stopwords.png')

stopwords.png

WordCloud 类参数小结

通过传入不同参数到 wc = WordCloud() ，我们可以轻松获得不同属性的词云：

至此，读者已经可以制作一些词云了，如果想做更好玩的词云，推荐阅读 「python词云高手进阶」

相关文章： 「python安装详细教程」

参考资料

• 官方文档:「WordCloud for Python documentation」
• b站:「词云可视化：四行Python代码轻松上手到精通」
• github:「amueller/word_cloud」