前段时间写了几篇关于生成词云的文章,其中包括:
Python 绘制悼念的词云蜡烛:https://pythondict.com/python-data-analyze/python-paint-mourn-candle/
等等,但是这几篇文章的转化对象都必须是多词汇组成的,也就是说他们生成词云的条件是词组必须够多,只有一两个词的话无法生成完整词云。
最近遇到许多朋友有特殊的需求,他们只想把 【名字】和【生日快乐】两个词组在一起,祝他人生日快乐,不想弄太复杂的东西,基于前面两篇文章是无法做到的。今天我们就来说说如何用这么少的词汇生成漂亮的词云。
1.准备
开始之前,你要确保Python和pip已经成功安装在电脑上噢,如果没有,请访问这篇文章:超详细Python安装指南进行安装。
Windows环境下打开Cmd(开始—运行—CMD),苹果系统环境下请打开Terminal(command+空格输入Terminal),准备开始输入命令安装依赖。
当然,我更推荐大家用VSCode编辑器,把本文代码Copy下来,在编辑器下方的终端装依赖模块,多舒服的一件事啊:Python 编程的最好搭档—VSCode 详细指南。
在终端输入以下命令安装我们所需要的依赖模块:
pip install wordcloud == 1.5.0pip install scipy == 1.1.0pip install multidict == 4.5.2pip install matplotlib == 2.2.4pip install fire == 0.2.1pip install numpy == 1.16.4
看到 Successfully installed xxx 则说明安装成功。或公众号后台回复生日快乐可获得本文全部代码,然后进入文件夹,输入一行命令安装所有依赖:
pip install -r requirements.txt
如果你不想折腾代码,安装完依赖后,输入以下命令就可以生成你的词云:
python birthday.py 图片位置 对象姓名
如:
python birthday.py example.png 宝典哥
2.编写代码
首先是引入词云对象,并初始化【生日快乐】和对方姓名:
细心的读者可能发现了,我们在这里用了MultiDict,这主要是因为wordcloud只允许接受【字典】数据结构,而Python内置的字典不允许重复值,所以我们只能引入multidict模块。
然后是插入新的生日快乐词云和对方姓名:
然后我们需要对图片进行一些处理,现在网络上的图片很多都包含一些杂色,因此需要把这些杂色去掉:
引入图片,去除杂色:
生成词云并渲染:
完整代码如下:
3.整合一句运行
接下来,我们使用上次提到的一行命令实现功能将这个功能打包成输入命令就能运行的程序,比如:
python birthday.py 图片 宝典哥
嘛,在完整代码最后面加一行语句就行了:
import firefire.Fire(gen_happy_birthday_cloud)
当然,别忘了还要import fire模块。最后实验一下:
python birthday.py example.png 宝典哥