三国演义词云的python代码
时间: 2023-07-10 16:24:28 浏览: 109
以下是使用Python生成三国演义词云的代码:
```python
# 导入需要的库
import jieba
import numpy as np
from PIL import Image
from wordcloud import WordCloud, ImageColorGenerator
import matplotlib.pyplot as plt
# 读取文本内容
with open('三国演义.txt', 'r', encoding='utf-8') as f:
text = f.read()
# 对文本进行分词处理
words = jieba.cut(text)
words = ' '.join(words)
# 读取背景图片
mask = np.array(Image.open('cloud.png'))
# 生成词云图
wc = WordCloud(font_path='simhei.ttf', background_color='white', max_words=2000, mask=mask)
wc.generate(words)
# 提取背景图片颜色
image_colors = ImageColorGenerator(mask)
# 绘制词云图
plt.imshow(wc)
plt.axis('off')
plt.show()
# 保存词云图
wc.to_file('wordcloud.png')
```
其中,`jieba` 库用于进行中文分词,`numpy` 和 `PIL` 库用于读取图片和生成图片数组,`wordcloud` 库用于生成词云图。需要将文本文件和背景图片放在同一目录下,并将背景图片命名为 `cloud.png`。生成的词云图将保存在当前目录下,并命名为 `wordcloud.png`。
阅读全文