让生成的词云图有个形状
时间: 2024-05-13 17:15:43 浏览: 114
词云背景图生成不同形状的图
5星 · 资源好评率100%
要让生成的词云图有一个特定的形状,你可以使用 Python 中的 wordcloud 库中的 mask 参数。该参数可以接受一个形状的图片,将词云图放置在该图片的形状内。
以下是一个示例代码,演示如何使用一个心形图片来制作词云图:
```python
from wordcloud import WordCloud, STOPWORDS
from PIL import Image
import numpy as np
# 读入文本文件
text = open("example.txt", encoding="utf-8").read()
# 读入形状图片
mask = np.array(Image.open("heart.png"))
# 设置停用词
stopwords = set(STOPWORDS)
stopwords.add("said")
# 生成词云图
wc = WordCloud(background_color="white", max_words=2000, mask=mask,
stopwords=stopwords, contour_width=3, contour_color='steelblue')
wc.generate(text)
# 保存词云图
wc.to_file("heart_wordcloud.png")
```
在上面的代码中,我们读入了一个文本文件,然后读入了一个心形图片作为词云图的形状。设置了一些停用词,然后使用 WordCloud 类生成词云图,并将其保存到文件中。
你可以使用类似的方法,使用你喜欢的形状图片来生成词云图。
阅读全文