使用wordcloud模块，绘制一个自定义的词云图形状

认真写程序的强哥

已于 2024-01-20 09:58:31 修改

阅读量2.1k

点赞数 10

文章标签： python Python学习 Python编程诗词云

于 2023-12-07 16:26:54 首次发布

本文链接：https://blog.csdn.net/qq_41314882/article/details/134858360

版权

在WordCloud模块中生成不同形状的词云图，可以通过使用mask功能来实现。首先，需要准备一张素材图片，如上图所示，网页搜的图片，有时间的话自己Ps扣一下背景图，这样显示的效果更好，该图片将被用作词云的形状。然后，使用imread函数从imageio库中导入该图片。接下来，使用WordCloud对象生成词云，然后使用to_file或to_array方法将生成的词云保存为图像。

涉及到的小知识点：

准备一个素材图片，背景为白色
open函数读取本地txt的方法
词云图相关模块的基本使用

01不按图片素材颜色生成词云图

# 导入词云制作库
import jieba
import wordcloud
import imageio

#读取本地图片，作为词云形状图片
mk = imageio.imread("刘亦菲.png")
w = wordcloud.WordCloud(mask=mk)

#构建词云对象w
w = wordcloud.WordCloud(background_color='white',
                        font_path='simkai.ttf',
                        mask=mk)

#读取txt文本数据
f = open('李白.txt',encoding='utf-8')
text= f.read()
text_list = jieba.lcut(text)
string = " ".join(text_list)

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

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

02按图片素材颜色生成词云图

import matplotlib.pyplot as plt
from wordcloud import WordCloud,ImageColorGenerator

# 1.定义文本内容
text='wdPython'
# 2.读取本地图片素材，作为词云形状图片
import imageio
mk = imageio.imread("刘亦菲.png")

# 3.构建词云对象w
wc = WordCloud(background_color="white",
               mask=mk,
               repeat=True,
               font_path='simkai.ttf')
# 4.将text字符串变量传入w的generate()方法，给词云输入文字
wc.generate(text)

# 调用wordcloud库中的ImageColorGenerator()函数，提取模板图片各部分的颜色
image_colors = ImageColorGenerator(mk)

# 5.显示默认词云图、按素材颜色的词云图和素材图片，按左、中、右显示
fig, axes = plt.subplots(1, 3)
# 最左边的是默认词云图
axes[0].imshow(wc)
# 中间的是按素材颜色的词云图，采用双线性插值的方法显示颜色
axes[1].imshow(wc.recolor(color_func=image_colors), interpolation="bilinear")
# 右边是素材图片
axes[2].imshow(mk, cmap=plt.cm.gray)
for ax in axes:
    ax.set_axis_off()
plt.show()

# 给词云对象按模板图片的颜色重新上色
wc_color = wc.recolor(color_func=image_colors)
# 6.将词云图片导出到当前文件夹
wc_color.to_file('词云图.png')