词云图python代码
时间: 2023-10-14 09:25:42 浏览: 61
基于Python实现的词云生成代码,从txt文件中读取词语,生成指定图片的词云图
生成词云图的Python代码如下:
```python
import wordcloud
import matplotlib.pyplot as plt
from PIL import Image
import numpy as np
import jieba
# 读取文本
with open('text.txt', 'r', encoding='utf-8') as f:
text = f.read()
# 对文本进行分词
words = jieba.cut(text)
# 过滤掉一些无用的词汇
stopwords = set()
with open('stopwords.txt', 'r', encoding='utf-8') as f:
for line in f.readlines():
stopwords.add(line.strip())
filtered_words = []
for word in words:
if word not in stopwords:
filtered_words.append(word)
# 拼接成字符串
text = ' '.join(filtered_words)
# 生成词云图
mask = np.array(Image.open('mask.png'))
wc = wordcloud.WordCloud(background_color='white', mask=mask, font_path='simsun.ttc')
wc.generate(text)
# 显示词云图
plt.imshow(wc)
plt.axis('off')
plt.show()
# 保存词云图
wc.to_file('wordcloud.png')
```
需要注意的是,代码中需要用到一个文本文件 `text.txt` 和一个停用词文件 `stopwords.txt`,还需要用到一个词云图形状的模板图片 `mask.png` 和一个中文字体文件 `simsun.ttc`,可以根据自己的需要进行替换。
阅读全文