妈妈再也不用担心我写专栏找不到合适的封面了!B站专栏的封面至少是我一直头疼的问题,每次写完文章却找不到合适的图片作为封面。
词云是一个很不错的选择,既美观,又提纲挈领。网上也有词云生成的工具,但大多收费/只能试用,很多设置也不能调整。于是我最终决定自己动手丰衣足食。
网上有很多关于词云生成的Python代码,但大多结构松散,要调一项设置要在代码里上上下下各个库里进行修改,不方便使用。于是我想把我们常用的设置封装到一个函数里,这样方便我们日常使用。
总的思路是采用Jieba库对中文文章进行词语拆分,使用Matplotlib库进行底层绘图支持,使用WoldCloud库对分词后的文章进行词频统计并画出词云。
原本的WordCloud库是针对英语所开发的,在英文里每个单词都是由空格分开的,所以只需统计一篇文章中单词的频率并画图即可。中文的词语则需要通过Jieba库来进行识别并分词,并再将文中的词语之间插入空格,才算完成对数据的准备工作,可以使用WordCloud库生成词云。WordCloud库默认生成的是充满整个画图区域的词云,效果不够美观。
一般的解决方案是选择一张已有的图片作为mask遮罩,使WordCloud在图形区域进行画图。在本项目中,预设了常用形状的遮罩(圆形、椭圆、矩形、多边形),通过matplotlib.patches模块,进行遮罩的生成,免去了手动上传遮罩图片的麻烦。
代码采用了函数的方式,对每个功能部分进行了封装,便于调用,也便于代码的理解。分成了make_mask生成遮罩图片;jieba_split使用jieba分词并插入空格;以及最后的cloud_generate生成WordCloud对象并调用Matplotlib绘图。
最终只需要调用cloud_generate()函数并填写相应的参数即可:
cloud_generate()中的参数:
text_addr:需要生成词云的文本文档的地址
shape:词云的形状
'circle' 或'c':圆形
'ellipse' 或'e':椭圆
'rectangle' 或'r':矩形
'square' 或's':正方形
整数n:正n边形
colormap:Matplotlib中的colormap数据对象,是颜色的组合部分Colormap预设
dpi:生成图片分辨率(默认为200)
output_addr:输出的图片的地址(默认为当前文件夹下wordcloud_output.png)
stopwords:列表,停止词(即你不希望出现在图片里的词)
例:
cloud_generate(text_addr = '.\\bili.txt', shape = 'e', colormap = 'Set2', dpi = 400, stopwords = ['进行', '我们'], output_addr = '.\\output.png')
即在当前目录下打开bili.txt作为要生成词云的文章,词云形状是椭圆,颜色组合是Set2,分辨率是400dpi,停止词是‘进行’和‘我们’(因为这两个词在文中出现的频率较高,但对文章内容的提示性不大,所以选择不生成在词云里),最后输出的图片在当前文件夹下,文件名为‘output.png’
最后,上代码~为了语法高亮,以截图的形式发上来,源码评论区自取~注:
1. 本项目依赖Jieba库、wordcloud库、matplotlib库。
2. 与程序文件同一文件夹下要有字体文件:“simhei.ttf”
3. 由于采用了函数的封装,在同一目录下的py文件可以直接将WordCloudGenerate.py当做库来导入,直接使用cloud_generate()函数from WordCloudGenerate import cloud_generate
也可以将WordCloudGenerate.py添加到python的库中,从任意地方调用此库。