几行代码教你使用python制作词云（四）

几行代码教你使用python制作词云（四）

这里给大家介绍wordcloud中勾勒轮廓参数，contour_width和contour_color设置轮廓宽度和颜色（文章最后附有案例完整代码）

1.准备图片和数据集

案例数据为蜡笔小新的介绍

图片也为蜡笔小新

2.代码分析

导入词云制作库wordcloud

# 导入词云制作库wordcloud
import wordcloud

将外部文件包含的文本保存在string变量中

# 将外部文件包含的文本保存在string变量中
string = open('./data/XiaoXin.txt',encoding='utf-8').read()

导入imageio库中的imread函数，并用这个函数读取本地图片，作为词云形状图片

# 导入imageio库中的imread函数，并用这个函数读取本地图片，作为词云形状图片
import imageio
mk = imageio.imread("./input/xiaoxin.jpg")

构建词云对象

# 构建词云对象w，注意增加参数contour_width和contour_color设置轮廓宽度和颜色
w = wordcloud.WordCloud(background_color="white",
                        mask=mk,
                        contour_width=1,
                        contour_color='steelblue')

将string变量传入w的generate()方法，给词云输入文字

 # 将string变量传入w的generate()方法，给词云输入文字
w.generate(string)

在控制台绘制并输出图像（这一步可要可不要）

# 在控制台绘制并输出图像
import matplotlib.pyplot as plt  #绘图
plt.imshow(w,interpolation="bilinear")
plt.axis("off")

将词云图片导出到指定文件夹

# 将词云图片导出到当前文件夹
w.to_file('./output/xiaoxin.png')

效果图

3.完整代码

# -*- coding: utf-8 -*-
"""
Created on Mon Jun  2 10:24:07 2020

@author: Jay-L
"""

# 导入词云制作库wordcloud
import wordcloud

# 将外部文件包含的文本保存在string变量中
string = open('./data/XiaoXin.txt',encoding='utf-8').read()

# 导入imageio库中的imread函数，并用这个函数读取本地图片，作为词云形状图片
import imageio
mk = imageio.imread("./input/xiaoxin.jpg")

# 构建词云对象w，注意增加参数contour_width和contour_color设置轮廓宽度和颜色
w = wordcloud.WordCloud(background_color="white",
                        mask=mk,
                        contour_width=1,
                        contour_color='steelblue')

# # 将string变量传入w的generate()方法，给词云输入文字
w.generate(string)

# 在控制台绘制并输出图像
import matplotlib.pyplot as plt  #绘图
plt.imshow(w,interpolation="bilinear")
plt.axis("off")

# 将词云图片导出到当前文件夹
w.to_file('./output/xiaoxin.png')