词云图python代码
时间: 2023-05-26 10:05:25 浏览: 66
基于Python实现的词云生成代码,从txt文件中读取词语,生成指定图片的词云图
以下是一个用Python生成词云图的示例代码:
```
import jieba
from wordcloud import WordCloud
import matplotlib.pyplot as plt
# 读取文本文件
with open('text.txt', 'r', encoding='utf-8') as f:
text = f.read()
# 对文本进行分词
word_list = jieba.cut(text)
# 将分词结果转换为字符串
words = ' '.join(word_list)
# 创建词云图对象
wordcloud = WordCloud(font_path='font.ttf', background_color='white', width=800, height=600)
# 生成词云图
wordcloud.generate(words)
# 显示词云图
plt.imshow(wordcloud)
plt.axis('off')
plt.show()
# 将词云图保存为图片文件
wordcloud.to_file('wordcloud.png')
```
其中,需要注意以下几点:
1. 需要先安装`jieba`和`wordcloud`两个库,可以使用pip安装。
2. `jieba`库是用于中文分词的,可以将文本分成一个个词语。
3. `WordCloud`是用于生成词云图的类,需要指定参数,例如`font_path`用于指定字体文件位置,`background_color`用于指定背景颜色,`width`和`height`用于指定图像宽度和高度。
4. `generate()`方法是用于生成词云图的,需要传入一个字符串参数,表示需要生成词云图的词语。
5. `imshow()`方法和`axis('off')`方法用于显示图像并隐藏坐标轴。
6. `to_file()`方法用于将词云图保存为文件。
阅读全文