python实现自定义图像词云的代码
时间: 2023-11-17 11:04:21 浏览: 96
使用python实现个性化词云的方法
以下是一个简单的Python代码,可以使用自定义图片生成词云:
```python
import jieba
from wordcloud import WordCloud, ImageColorGenerator
import matplotlib.pyplot as plt
from PIL import Image
# 读取文本
with open('text.txt', 'r', encoding='utf-8') as f:
text = f.read()
# 分词
text = ' '.join(jieba.cut(text))
# 读取图片
image = Image.open('image.png')
# 生成词云
wordcloud = WordCloud(background_color='white', mask=image).generate(text)
# 从图片中提取颜色
image_colors = ImageColorGenerator(image)
# 显示词云
plt.imshow(wordcloud.recolor(color_func=image_colors))
plt.axis('off')
plt.show()
# 保存词云
wordcloud.to_file('wordcloud.png')
```
其中,`text.txt`是要生成词云的文本文件,`image.png`是要使用的图片文件。可以根据自己的需求修改文件名和路径。生成的词云将会保存在当前目录下的`wordcloud.png`文件中。
阅读全文