Python词云图制作教程：wordcloud库安装与使用详解

作者是本科在读的人工智能专业学生，喜爱计算机和编程，会通过写博客来记录自己的学习历程。

个人主页：小嗷犬的博客

个人的信条是：为天地确立起生生之心，为百姓确立起安身立命之命，为过往的圣贤继承断绝的学问，为万代开创太平之世。

本文内容：Python 词云图：wordcloud库的使用

Python 词云图：wordcloud库的使用

wordcloud库的安装

wordcloud 库是 Python 中的一个优秀的第三方词云展示函数库。它能够利用文本来生成词云图。并且需要通过 pip 指令来进行安装。

pip install wordcloud

wordcloud库把词云当作一个WordCloud对象

wordcloud库的使用常用函数方法

- get_single_color_func(color)：获取单个颜色设置函数，color 为指定颜色。

函数描述

wordcloud.WordCloud()

根据参数配置词云图对象

w.generate(txt)

向WordCloud对象中加载文本txt（会根据空格分词）

w.to_file(filename)

将词云输出为图像文件，.png或.jpg格式

WordCloud对象常用配置参数

WordCloud对象常用配置参数如下：

参数描述

width

指定词云对象生成图片的宽度，默认400像素

height

指定词云对象生成图片的高度，默认200像素

min_font_size

指定词云中字体的最小字号，默认4号

max_font_size

指定词云中字体的最大字号，根据高度自动调节

font_step

指定词云中字体字号的步进间隔，默认为1

font_path

指定字体文件的路径，默认None

max_words

指定词云显示的最大单词数量，默认200

stopwords

指定词云的排除词列表，即不显示的单词列表

mask

指定词云形状，默认为长方形，需要引用imread()函数

background_color

指定词云图片的背景颜色，默认为黑色

colormap

指定词云文字的配色集，默认为’viridis’

配色集

常用配色集如下：

生成词云图

结合之前的那篇博客，即关于 Python 中文分词中 jieba 库使用的博客，我们便可以着手尝试生成第一张词云图了。

import[id_1110806745]
import wordcloud
from imageio import imread
# 排除词库
excludes = {
    '两个', '一个', '只见', '如何', '那里', '说道', '这里', '出来', '这个', '今日', '便是', '问道',
    '起来', '甚么', '因此', '却是', '我们', '正是', '三个', '如此', '且说', '不知', '不是', '只是',
    '次日', '不曾', '不得', '一面', '看时', '不敢', '如今', '来到', '当下', '原来', '喝道', '只得',
    '里面', '大喜', '一齐', '商议', '那个', '公人', '将来', '前面', '那厮', '城中', '下山', '不见',
    '怎地', '上山', '随即', '不要'
}
# 读入水浒传，分词，并以空格连接
txt = open("Documents/《水浒传》.txt", "r", encoding='utf-8').read()
words = jieba.lcut(txt)
txt0 = ''
for i in words:
    if len(i) > 1:
        txt0 += ' '
        txt0 += i
txt0.replace('宋江道', '宋江')  # 纠正错误分词
mk = imread('皮卡丘.jpg')  # 设置蒙版为皮卡丘
w = wordcloud.WordCloud(
    width=1920,
    height=1080,  # 设置图片长宽为1080p
    background_color='white',  # 设置背景颜色为白色
    font_path=C 盘下 Windows 文件夹中的 Fonts 文件夹里有 msyh.ttc 文件。,  # 设置字体为微软雅黑
    max_words=300,  # 设置词汇最大数量为300
    stopwords=excludes,  # 设置排除词库
    mask=mk,  # 设置蒙版
    colormap='magma'  # 设置配色集为magma
)
w.generate(txt0)
w.to_file('img.png')