绘制词云图的Python代码
时间: 2023-06-26 10:07:17 浏览: 34
生成词云图需要用到Python中的`wordcloud`库,可以通过以下代码安装:
```
pip install wordcloud
```
然后,你可以使用以下代码生成词云图:
```python
import wordcloud
from wordcloud import WordCloud, STOPWORDS
import matplotlib.pyplot as plt
# 读取文本文件
text = open('example.txt', 'r').read()
# 创建停用词列表
stopwords = set(STOPWORDS)
stopwords.update(['said', 'will', 'one', 'now', 'well', 'man', 'may', 'us', 'say', 'made', 'much', 'must', 'good', 'great'])
# 创建词云对象并生成词云
wc = WordCloud(width=800, height=800, background_color='white', stopwords=stopwords, min_font_size=10).generate(text)
# 显示词云图
plt.figure(figsize=(8, 8), facecolor=None)
plt.imshow(wc)
plt.axis('off')
plt.tight_layout(pad=0)
plt.show()
```
上述代码中,我们首先读取了一个文本文件`example.txt`,然后创建一个停用词列表,过滤掉一些常用词,接着创建词云对象并生成词云图,最后通过`matplotlib`库将词云图进行显示。