ITPub博客

首页 > 应用开发 > IT综合 > 【Python】Wordcloud 词云

【Python】Wordcloud 词云

原创 IT综合 作者:小亮520cl 时间:2017-10-19 11:44:41 0 删除 编辑

本周为大家带来炫酷好玩的 wordcloud 词云构造库。

使用 wordcloud 可以做出这样的图片:

还可以做出这样的:

接下来,我们来学习如何制作属于自己的词云图。

本来想说一句,安装过程不表,直接进入正题,可是在编程教室的QQ群和微信群中都有人遇到wordcloud 安装失败的问题。

所以在此简单提一下安装过程:

当然最直接的是 pip

pip install wordcloud

但很多同学会发现直接 pip 是会报错的,所以这里提供第二种安装方式:

下载 whl 手动安装包:http://www.lfd.uci.edu/~gohlke/pythonlibs/#wordcloud

静静等待页面加载完成,然后选择适合自己 Python 的 wordcloud 版本,最后运行以下命令。filepath 替换为你的安装文件完整路径。

pip install filepath\wordcloud-1.2.1-cp35-cp35m-win_amd64.whl

wordcloud 在安装过程中会自动安装其依赖的库。

基本的用法

我们读入一本英文小说,取其中的词汇生成一张词云图

  1. # 导入 wordcloud 模块和 matplotlib 模块
  2. from wordcloud import WordCloud
  3. import matplotlib.pyplot as plt
  4. # 读入一个txt文件
  5. text = open('Jane Eyre.txt','r').read()
  6. # 生成词云
  7. wordcloud = WordCloud().generate(text)
  8. # 显示词云图片
  9. plt.imshow(wordcloud)
  10. plt.axis('off')
  11. plt.show()
  12. # 保存图片
  13. wordcloud.to_file('test.jpg')


结果如图所示

运行脚本之后就可以看到生成的图片了,是不是很简单。本文主要讲 wordcloud。代码中涉及到的 matplotlib 就不做更详细解释。

除了直接读入文本生成词云,也可以使用字典格式的词频作为输入

  1. # 读入一组词频字典文件
  2. text_dict = {
  3.     'you': 2993,
  4.     'and': 6625,
  5.     'in': 2767,
  6.     'was': 2525,
  7.     'the': 7845,
  8. }
  9. wordcloud = WordCloud().generate_from_frequencies(text_dict)


还可以将词云填充到指定的形状之中。为达到填充指定形状的效果,需要使用 png 格式的图片。

  1. # 读入图片
  2. from scipy.misc import imread
  3. bg_pic = imread('Anne_Hathaway.png')
  4. # 配置词云参数
  5. wc = WordCloud(
  6.     # 设置字体
  7.     font_path = 'BeaverScratches.ttf',
  8.     # 设置背景色
  9.     background_color='white',
  10.     # 允许最大词汇
  11.     max_words=200,
  12.     # 词云形状
  13.     mask=bg_pic,
  14.     # 最大号字体
  15.     max_font_size=100,
  16. )
  17. # 生成词云
  18. wc.generate(text)
  19. # 保存图片
  20. wc.to_file('word.jpg')


最后的效果如下

还可以结合 jieba分词,实现用中文填充的词云图(如果不通过分词,无法直接生成正确的中文词云)。使用中文填充时一定得指定中文的字体,否则会出现乱码。如果报错找不到字体,就复制一个中文字体文件放在代码目录下。

  1. #-*- coding:utf-8 -*-
  2. from scipy.misc import imread
  3. import matplotlib.pyplot as plt
  4. from wordcloud import WordCloud
  5. import jieba
  6. from collections import Counter
  7. # 读入 西游记 txt 文件,windows 下过滤编码错误
  8. text = open('西游记.txt',encoding='utf-8',errors='ignore').read()
  9. # 使用 jieba 分词
  10. text_jieba = list(jieba.cut(text))
  11. # 使用 counter 做词频统计,选取出现频率前 100 的词汇
  12. c = Counter(text_jieba)
  13. common_c = c.most_common(100)
  14. # 读入图片
  15. bg_pic = imread('Anne_Hathaway.png')
  16. # 配置词云参数
  17. wc = WordCloud(
  18.     # 设置字体
  19.     font_path = '李旭科书法1.4.ttf',
  20.     # 设置背景色
  21.     background_color='white',
  22.     # 允许最大词汇
  23.     max_words=200,
  24.     # 词云形状
  25.     mask=bg_pic,
  26.     # 最大号字体
  27.     max_font_size=100,
  28. )
  29. # 生成词云
  30. wc.generate_from_frequencies(dict(common_c))
  31. # 生成图片并显示
  32. plt.figure()
  33. plt.imshow(wc)
  34. plt.axis('off')
  35. plt.show()
  36. # 保存图片
  37. wc.to_file('anne.jpg')


最终的结果:


关于 wordcloud 的一些基本的常用的方法就在这里。想了解更多特性的同学,可以访问官方网站: https://amueller.github.io/word_cloud/  

来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/29096438/viewspace-2146151/ ,如需转载,请注明出处,否则将追究法律责任。

请登录后发表评论 登录
全部评论
毕业以后专业任职数据库工程师职位,itpub一直作为自己的笔记记录的地方,blog写的不详细,草书笔记,仅供参考!

注册时间:2013-09-12

  • 博文量
    529
  • 访问量
    1126931