运用wordcloud库将图片转化成词云图
时间: 2023-09-06 11:13:05 浏览: 106
好的,你需要先安装 `wordcloud` 库,可以通过以下命令进行安装:
```
pip install wordcloud
```
安装完成后,你可以使用下面的代码将图片转化成词云图:
```python
from wordcloud import WordCloud, ImageColorGenerator
import matplotlib.pyplot as plt
from PIL import Image
import numpy as np
import jieba
# 读取文本,使用 jieba 分词
text = open('example.txt', 'r', encoding='utf-8').read()
cut_text = jieba.cut(text)
result = ' '.join(cut_text)
# 读取图片
image = Image.open('example.png')
graph = np.array(image)
# 生成词云图
wc = WordCloud(font_path='simhei.ttf', background_color='white', max_words=2000, mask=graph)
wc.generate(result)
# 根据图片颜色生成词云图颜色
image_color = ImageColorGenerator(graph)
wc.recolor(color_func=image_color)
# 显示词云图
plt.imshow(wc)
plt.axis('off')
plt.show()
# 保存词云图
wc.to_file('example_wordcloud.png')
```
其中,你需要将 `example.txt` 替换成你要生成词云图的文本,将 `example.png` 替换成你要生成词云图的图片。需要注意的是,`simhei.ttf` 是中文字体,如果你要生成英文词云图可以将其删除或替换成其他字体。最后,词云图会显示在屏幕上并保存为 `example_wordcloud.png`。
阅读全文