wordcloud词云图美化

前言

之前的文章我们已经介绍了如何使用wordcloud库制作中英文词云图，并介绍了中英文停用词的使用方法，但如何美化词云图，例如换字体背景颜色，背景换成图片等，这些将在本篇文章进行详细介绍。

参数详解

要想美化词云图，首先就要明白wordcloud自定义的参数的各个意思。

font_path : string  #字体路径，需要展现什么字体就把该字体路径+后缀名写上，如：font_path = '黑体.ttf'

width : int (default=400) #输出的画布宽度，默认为400像素

height : int (default=200) #输出的画布高度，默认为200像素

prefer_horizontal : float (default=0.90) #词语水平方向排版出现的频率，默认 0.9 （所以词语垂直方向排版出现频率为 0.1 ）

mask : nd-array or None (default=None) #如果参数为空，则使用二维遮罩绘制词云。如果 mask 非空，设置的宽高值将被忽略，遮罩形状被 mask 取代。除全白（#FFFFFF）的部分将不会绘制，其余部分会用于绘制词云。如：bg_pic = imread('读取一张图片.png')，背景图片的画布一定要设置为白色（#FFFFFF），然后显示的形状为不是白色的其他颜色。可以用ps工具将自己要显示的形状复制到一个纯白色的画布上再保存，就ok了。

scale : float (default=1) #按照比例进行放大画布，如设置为1.5，则长和宽都是原来画布的1.5倍

min_font_size : int (default=4) #显示的最小的字体大小

font_step : int (default=1) #字体步长，如果步长大于1，会加快运算但是可能导致结果出现较大的误差

max_words : number (default=200) #要显示的词的最大个数

stopwords : set of strings or None #设置需要屏蔽的词，如果为空，则使用内置的STOPWORDS

background_color : color value (default=”black”) #背景颜色，如background_color='white',背景颜色为白色

max_font_size : int or None (default=None) #显示的最大的字体大小

mode : string (default=”RGB”) #当参数为“RGBA”并且background_color不为空时，背景为透明

relative_scaling : float (default=.5) #词频和字体大小的关联性

color_func : callable, default=None #生成新颜色的函数，如果为空，则使用 self.color_func

regexp : string or None (optional) #使用正则表达式分隔输入的文本

collocations : bool, default=True #是否包括两个词的搭配

colormap : string or matplotlib colormap, default=”viridis” #给每个单词随机分配颜色，若指定color_func，则忽略该方法

random_state : int or None  #为每个单词返回一个PIL颜色

我们就先简单举个例子，我们做中文词云，必须要设置中文字体，不然就会乱码，所以必须设置font_size；默认背景是黑色的，我们要设置为其他颜色就需要设置background_color；我们再将长宽比例设置为16：9，就要设置width和height；当然设置停用词就需要使用stopwords。

from matplotlib import pyplot as plt
from wordcloud import WordCloud
import jieba

text = '我叫罗攀，他叫关羽，我叫罗攀，他叫刘备'
cut_word = " ".join(jieba.cut(text))

stopwords = set()
content = [line.strip() for line in open('hit_stopwords.txt','r').readlines()]
stopwords.update(content)

wc = WordCloud(background_color = '#F3F3F3',
               font_path = r'/System/Library/Fonts/Supplemental/Songti.ttc',
               stopwords = stopwords,
               width = 1600,
               height = 900)
wc.generate(cut_word)

plt.imshow(wc, interpolation='bilinear')
plt.axis('off') #取消坐标轴
plt.show()

自定义背景颜色

最后，就是大家最关心的，如何设置背景颜色的问题，其实就是用mask参数就行，我们再网上下载了一个图片。

from matplotlib import pyplot as plt
from wordcloud import WordCloud,ImageColorGenerator
import jieba

text = '我叫罗攀，他叫关羽，我叫罗攀，他叫刘备，aka，王五，李四'
cut_word = " ".join(jieba.cut(text))

stopwords = set()
content = [line.strip() for line in open('hit_stopwords.txt','r').readlines()]
stopwords.update(content)

background_image = plt.imread('love.jpeg')

wc = WordCloud(background_color = 'white',
               font_path = r'/System/Library/Fonts/Supplemental/Songti.ttc',
               stopwords = stopwords,
               mask = background_image)
wc.generate(cut_word)

#改变字体颜色
img_colors = ImageColorGenerator(background_image)
#字体颜色为背景图片的颜色
wc.recolor(color_func=img_colors)

plt.imshow(wc, interpolation='bilinear')
plt.axis('off') #取消坐标轴
plt.show()

本期分享到此结束，我们下期再见~