用python的wordcloud库写一个傲慢与偏见的词云图代码
时间: 2023-06-21 11:20:15 浏览: 96
好的,这里是基于Python的wordcloud库实现傲慢与偏见的词云图的代码:
```python
import numpy as np
import pandas as pd
from PIL import Image
from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator
import matplotlib.pyplot as plt
# 读取文本文件
with open('pride_and_prejudice.txt', 'r', encoding='utf-8') as f:
text = f.read()
# 读取背景图
mask = np.array(Image.open('heart.png'))
# 设置停用词
stopwords = set(STOPWORDS)
stopwords.add('said')
# 生成词云图
wordcloud = WordCloud(background_color='white',
mask=mask,
stopwords=stopwords,
max_font_size=200,
max_words=200,
width=1200,
height=800,
font_path='msyh.ttc').generate(text)
# 根据背景图颜色生成相应的颜色
image_colors = ImageColorGenerator(mask)
plt.figure(figsize=(12, 8))
plt.imshow(wordcloud.recolor(color_func=image_colors), interpolation="bilinear")
plt.axis("off")
plt.show()
```
这里我们使用了一张心形图片作为背景图,可以自己在网上找一张喜欢的图片作为背景。同时,我们还可以根据背景图的颜色生成相应的颜色,使词云图更加美观。
阅读全文