python做词云的图片素材_还在为专栏封面发愁?我用Python写了个词云生成器!

妈妈再也不用担心我写专栏找不到合适的封面了!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的库中,从任意地方调用此库。

你可能感兴趣的:(python做词云的图片素材)