python中wordcloud库方法介绍
时间: 2023-12-01 10:02:45 浏览: 70
Python中的WordCloud库是一个用于可视化文本数据的工具。它可以根据文本中单词出现的频率和重要性将单词云展现出来,从而更加直观地展示文本数据的特点和关键词。
使用WordCloud库,需要先安装它,可以通过pip安装:
```
pip install wordcloud
```
接下来,就可以使用WordCloud库的核心函数WordCloud()来生成单词云。WordCloud()函数的主要参数有:
- width和height:单词云的宽度和高度;
- background_color:单词云的背景颜色;
- stopwords:需要屏蔽的单词,多个单词之间用空格隔开;
- max_words:单词云显示的最大单词数;
- mask:单词云形状,默认为矩形;
- contour_width和contour_color:单词云边框线的宽度和颜色;
- font_path:字体文件路径。
生成单词云的步骤如下:
1. 读取文本数据,并进行分词处理;
2. 去除停用词等无用词汇;
3. 计算每个单词的词频;
4. 根据词频和重要性生成单词云。
代码示例:
```python
import jieba
from wordcloud import WordCloud
# 读取文本数据
with open('data.txt', 'r', encoding='utf-8') as f:
text = f.read()
# 分词
words = jieba.cut(text)
# 去除停用词
stopwords = {'的', '了', '是', '和', '在', '有', '不', '这', '中', '他'}
words = [word for word in words if word not in stopwords]
# 计算词频
word_counts = {}
for word in words:
if word not in word_counts:
word_counts[word] = 1
else:
word_counts[word] += 1
# 生成单词云
wc = WordCloud(font_path='msyh.ttc', width=800, height=600, background_color='white',
max_words=200, contour_width=1, contour_color='steelblue')
wc.generate_from_frequencies(word_counts)
wc.to_file('wordcloud.png')
```
该代码实现的功能是读取文本数据,进行分词处理,去除停用词,计算每个单词的词频,并生成一个包含200个单词的单词云。生成的单词云会保存为一个PNG图片。
阅读全文