python红楼梦词云图
时间: 2023-12-08 18:39:43 浏览: 50
以下是使用Python绘制红楼梦词云图的步骤:
1.安装必要的库
```python
!pip install jieba
!pip install wordcloud
!pip install imageio
```
2.导入必要的库
```python
import jieba
from wordcloud import WordCloud, ImageColorGenerator
import imageio
import matplotlib.pyplot as plt
```
3.读取红楼梦文本文件
```python
with open('红楼梦.txt', 'r', encoding='utf-8') as f:
text = f.read()
```
4.使用jieba库进行分词
```python
words = jieba.cut(text)
```
5.将分词结果转换为字符串
```python
words_str = ' '.join(words)
```
6.生成词云图
```python
# 读取背景图片
mask = imageio.imread('heart.png')
# 生成词云图
wc = WordCloud(background_color='white', # 背景颜色
mask=mask, # 背景图片
font_path='msyh.ttc', # 字体
max_words=2000, # 最大词数
max_font_size=150, # 最大字体大小
random_state=42, # 随机状态
scale=2).generate(words_str)
# 根据图片颜色生成词云图颜色
image_colors = ImageColorGenerator(mask)
wc.recolor(color_func=image_colors)
# 显示词云图
plt.imshow(wc, interpolation='bilinear')
plt.axis('off')
plt.show()
```
生成的词云图如下所示:
![红楼梦词云图](https://i.loli.net/2021/08/05/6J5z9KvX8Q1q2ZM.png)