根据词频、背景图绘制词云图

根据词频、背景图绘制词云图

记录一下绘制词云图的过程

首先说明一下,绘制词云图完整的过程应如下:

  • 获取一段文本,也就是要根据这个文本来绘制词云图
  • 制作停用词词典,简单来说,停用词就是分词的规则,例如,可以根据一些副词来断开文本进行分词
  • 根据分词的结果统计词频,也就是高频词
  • 设置词云图背景
  • 绘制词云图

我现在跳过分词的过程,直接通过词频来绘制词云图

直接上代码

import numpy as np
from wordcloud import WordCloud
from PIL import Image
import matplotlib.pyplot as plt
import faker
from collections import Counter

fake = faker.Faker(locale='zh_CN') 
city = []
for i in range(100000):
    city.append(fake.city())
dic = Counter(city)
dic = dict(dic)

background = "bus.png"
mask = np.array(Image.open(background))
# mask = imread(background)

wc = WordCloud(
    font_path='C:/Windows/Fonts/simfang.ttf',   # 设置字体(这里选择“仿宋”)
    background_color='white',                   # 背景颜色
    mask=mask,                                  # 文字颜色+形状(有mask参数再设定宽高是无效的)
    max_words=200,                              # 显示词数
    max_font_size=150,                          # 最大字号
    contour_color="steelblue",
    contour_width=1                             # 图片轮廓线
)

wc.generate_from_frequencies(dic)  
plt.figure('词云')                                                               # 弹框名称与大小
plt.subplots_adjust(top=0.99,bottom=0.01,right=0.99,left=0.01,hspace=0,wspace=0) # 调整边距
plt.imshow(wc, cmap=plt.cm.gray, interpolation='bilinear')                       # 处理词云
plt.axis('off')                    
plt.show()  

先看看效果,再解释代码

根据词频、背景图绘制词云图_第1张图片

注意我仍然使用faker生成的假数据,使用collections.Counter方法来统计词频,并强制转换成字典格式的数据

background = "bus.png"
mask = np.array(Image.open(background))

上面这两行代码是在设置图片背景,这里比较关键,和最终展示效果有很大关系

首先看看原图
根据词频、背景图绘制词云图_第2张图片

这是我在网上找的一张图片,还有水印和背景色,如果直接使用这张图片作为蒙版,看看是什么效果

根据词频、背景图绘制词云图_第3张图片

这根本看不出来是一辆车,就是因为有背景色导致的。

该如何处理呢,有很多种方式,博主是在ps中使用了魔棒工具删除了背景,得到了下面这张无背景的图片

根据词频、背景图绘制词云图_第4张图片

然后设置为背景

wc = WordCloud(
    font_path='C:/Windows/Fonts/simfang.ttf',   # 设置字体(这里选择“仿宋”)
    background_color='white',                   # 背景颜色
    mask=mask,                                  # 文字颜色+形状(有mask参数再设定宽高是无效的)
    max_words=200,                              # 显示词数
    max_font_size=150,                          # 最大字号
    contour_color="steelblue",
    contour_width=1                             # 图片轮廓线
)

上面是绘制背景图的代码

最后一行代码很关键,contour_width=1用它来设置图片的轮廓线,也就是背景图片的轮廓线,可以很清楚的展示出来图片的轮廓,如果不写,背景图也会不理想

上面是绘制背景图的代码

最后一行代码很关键,contour_width=1用它来设置图片的轮廓线,也就是背景图片的轮廓线,可以很清楚的展示出来图片的轮廓,如果不写,背景图也会不理想

你可能感兴趣的:(Python基础,python,开发语言)