大数据

利用python制作词云图

2019-08-04  本文已影响1人  言木石

在统计线上商城用户所购买物品的喜好时,考虑到以词云图的形式来作展示,通过查找网上相关资料得以实现,此处以郁达夫《故都的秋》中的文章来作示例。
1.需要调用的包主要有wordcloud,jieba,可直接利用pip install wordcloud、pip install jieba进行安装;


wordcloud版本示例 jieba版本示例

2.关于字体设置,可在C:\Windows\Fonts下,右键勾选属性,查看对应字体的字体名称;


字体属性图示

3.关于是否调用Jieba,目前经过测试发现并无什么差别,具体应用场景后续再逐渐摸索;

4.如果要让词云生成特定图片样式,则导入PIL.Image及numpy,然后再作相应处理,详见下文代码。

代码模块1-不带图片样式:

from wordcloud import WordCloud #导入wordcloud包
with open("故都的秋.txt" ,encoding="utf-8")as file: #打开文本内容文件
    text=file.read()    #读取文本内容
    wordcloud=WordCloud(
        font_path="C:/Windows/Fonts/msyh.ttc",  #字体路径,需要展现什么字体就把该字体路径+后缀名写上,如:font_path = ‘黑体.ttf’
        background_color="white",   #背景颜色
        prefer_horizontal = 1,  #词语水平方向排版出现的频率,如果为1这说明均为水平方向排版,垂直方向排版则为0,反之亦然。
        width=600,  #输出图片的宽度
        height=300, #输出图片的高度
        max_words=50    #显示的词的最大个数
    ).generate(text)
    image=wordcloud.to_image()  #生成图片
    image.show()    #显示图片
生成的词云图

代码模块2-带图片样式:
假设我们想让词云图最终呈现下图所示枫叶样式


fengye.jpg
from wordcloud import WordCloud #导入wordcloud包
import numpy    #导入Numpy包
import PIL.Image as Image   #导入图像处理库PIL的Image模块,该模块的运用详见https://blog.csdn.net/youzhouliu/article/details/81044465
with open("故都的秋.txt" ,encoding="utf-8")as file: #打开文本内容文件
  text=file.read()    #读取文本内容
  mask_pic = numpy.array(Image.open("fengye.jpg"))    #将图片像素作矩阵式处理,便于后续词云图与底层图融合使用。
  wordcloud=WordCloud(
      font_path="C:/Windows/Fonts/msyh.ttc",  #字体路径,需要展现什么字体就把该字体路径+后缀名写上,如:font_path = ‘黑体.ttf’
      background_color="white",   #背景颜色
      width=600,  #设置图片的宽度
      height=300, #设置图片的高度
      max_words=50,   #设置显示词的最大个数
      mask=mask_pic   #如果参数为空,则使用二维遮罩绘制词云。如果 mask 非空,设置的宽高值将被忽略,遮罩形状被 mask 取代。
  ).generate(text)
  image=wordcloud.to_image()  #生成图片
  image.show()    #显示图片
生成的词云图

附录:wordcloud的参数说明

参考文档:
Python 生成词云图
Python图像处理库PIL的Image模块介绍
用分割点坐标将图像二值化
mask图像

上一篇 下一篇

猜你喜欢

热点阅读