通过Python的wordcloud库将单词生成词云(心形形状)

前言

大家好，我是空空star，本篇给大家分享一下通过Python的wordcloud库将单词生成词云(心形形状)。

一、wordcloud库是什么？

Python的wordcloud库是一个用于生成词云的Python包。它可以将一段文本中出现频率高的单词按其出现频率大小以及颜色深浅排列成一个词云图形，从而更好地展示文本中的信息。
使用wordcloud库，你可以指定词云的形状、颜色、字体等参数，以及控制词云中单词的排列方式、大小等属性。你可以使用wordcloud库来生成各种类型的词云，例如词频统计、情感分析等。

二、安装wordcloud库

pip install wordcloud

三、查看wordcloud库版本

 pip show wordcloud

Name: wordcloud
Version: 1.8.2.2
Summary: A little word cloud generator
Home-page: https://github.com/amueller/word_cloud
Author: Andreas Mueller
Author-email: [email protected]
License: MIT
Requires: matplotlib, numpy, pillow
Required-by:

四、使用方法

1.引入库

import jieba
import numpy as np
from PIL import Image
from wordcloud import WordCloud

2.定义图片路径

local = '/Users/kkstar/Downloads/video/pic/'

3.定义需要分词的文本

text = "大家好我是空空star我爱发动态我喜欢使用搜索引擎模式进行分词"

4.采用jieba搜索引擎模式分词

cut_text = " ".join(jieba.cut_for_search(text))

5.加载心形图片

mask = np.array(Image.open(local+"heart.png"))

6.创建词云对象

wc = WordCloud(font_path="STHeitiMedium.ttc",  # 设置字体
               background_color='white',  # 设置背景颜色
               mask=mask,  # 设置背景图片
               max_words=2000,  # 设置最大显示的词数
               max_font_size=100,  # 设置字体最大值
               width=500,  # 设置画布的宽度
               height=500,  # 设置画布的高度
               )

7.生成词云

wc.generate(cut_text)

8.保存词云图

wc.to_file(local+"wordcloud.png")

9.词云图效果

总结

要设置WordCloud的心形形状，需要使用mask参数来指定心形图片的路径，使用Python的Pillow库加载心形图片，并将其传递给WordCloud对象。

jieba库和pillow库介绍：
通过Python的jieba库对文本进行分词
通过Python的PIL库给图片添加文本水印