python作爱心词云图
时间: 2023-11-03 19:59:10 浏览: 48
以下是一个简单的Python代码,用于生成一个爱心形状的词云图:
```python
import numpy as np
from PIL import Image
from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator
import matplotlib.pyplot as plt
# 读入文本
text = open('love.txt', 'r', encoding='utf-8').read()
# 读入图片
mask = np.array(Image.open('heart.png'))
# 设置停用词
stopwords = set(STOPWORDS)
stopwords.add('said')
# 设置词云参数
wc = WordCloud(background_color='white',
mask=mask,
max_words=200,
stopwords=stopwords,
font_path='msyh.ttc',
random_state=42)
# 生成词云
wc.generate(text)
# 基于图片的颜色生成字体颜色
image_colors = ImageColorGenerator(mask)
wc.recolor(color_func=image_colors)
# 显示词云
plt.imshow(wc, interpolation='bilinear')
plt.axis('off')
plt.show()
```
其中,需要准备一个文本文件 `love.txt` 和一个心形图片 `heart.png`。在代码中,我们通过 `WordCloud` 类来生成词云,并设置了一些参数,例如背景色、掩膜形状、最大词数、停用词等。最后通过 `imshow` 函数显示词云图。你可以根据自己的需求调整代码中的参数。