Python如何生成词云

前言

今天教大家用wrodcloud模块来生成词云,我读取了一篇小说并生成了词云,先看一下效果图:

效果图一:

效果图二:

根据效果图分析的还是比较准确的,小说中的主人公就是“程理”,所以出现次数最多。图中有两种模式,一种是默认的模式,另一种是自己添加图片作为背景。下面我就带大家一起来学习怎样去生成词云!

wordcloud的安装

对于新人来说安装wordcloud模块就是一大关,我们一般都是通过Pycharm或者PIP安装的,但是在安装wordcloud时会提示错误,如下:

我的解决方法是直接访问下面的网址,根据你的电脑和py版本下载对应的whl文件

https://www.lfd.uci.edu/~gohlke/pythonlibs/#wordcloud

下载完成后,把whl文件放在pip路径下,输入下面的代码,记住代码的whl文件名一定是适合你的文件名,我只是举例:

pipinstallwordcloud-1.4.1-cp36-cp36m-win32.whl

运行后会显示安装成功

wordcloud的简单用法

wordcloud的基本方法,看源码如下:

各个参数大家应该一看就明白,另外wordcloud还有几个其他方法,比如fit_words,generate(text),to_file(filename)等等都是非常好理解的。

wordcloud生成词云

效果图一的代码:

fromwordcloudimportWordCloud

importmatplotlib.pyplotasplt#绘制图像的模块

importjieba#jieba分词

path_txt='C://Users/Administrator/Desktop/all.txt'

f = open(path_txt,'r',encoding='UTF-8').read()

# 结巴分词,生成字符串,wordcloud无法直接生成正确的中文词云

cut_text =" ".join(jieba.cut(f))

wordcloud = WordCloud(

#设置字体,不然会出现口字乱码,文字的路径是电脑的字体一般路径,可以换成别的

font_path="C:/Windows/Fonts/simfang.ttf",

#设置了背景,宽高

background_color="white",width=1000,height=880).generate(cut_text)

plt.imshow(wordcloud, interpolation="bilinear")

plt.axis("off")

plt.show()

运行后:

效果图二,给词云加上图片背景,代码如下:

fromPILimportImage

fromwordcloudimportWordCloud, ImageColorGenerator

importmatplotlib.pyplotasplt

importnumpyasnp

importjieba

defGetWordCloud():

path_txt ='C://Users/Administrator/Desktop/all.txt'

path_img ="C://Users/Administrator/Desktop/timg.jpg"

f = open(path_txt,'r', encoding='UTF-8').read()

background_image = np.array(Image.open(path_img))

# 结巴分词,生成字符串,如果不通过分词,无法直接生成正确的中文词云,感兴趣的朋友可以去查一下,有多种分词模式

#Python join() 方法用于将序列中的元素以指定的字符连接生成一个新的字符串。

cut_text =" ".join(jieba.cut(f))

wordcloud = WordCloud(

# 设置字体,不然会出现口字乱码,文字的路径是电脑的字体一般路径,可以换成别的

font_path="C:/Windows/Fonts/simfang.ttf",

background_color="white",

# mask参数=图片背景,必须要写上,另外有mask参数再设定宽高是无效的

mask=background_image).generate(cut_text)

# 生成颜色值

image_colors = ImageColorGenerator(background_image)

# 下面代码表示显示图片

plt.imshow(wordcloud.recolor(color_func=image_colors), interpolation="bilinear")

plt.axis("off")

plt.show()

if__name__ =='__main__':

GetWordCloud()

运行后:

如果有不懂的朋友可以直接在公众号上加我的微信,希望对大家有所帮助!

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20180517G17N0Z00?refer=cp_1026
  • 腾讯「云+社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 yunjia_community@tencent.com 删除。

扫码关注云+社区

领取腾讯云代金券