p y t hon词云
时间: 2023-12-14 08:35:38 浏览: 110
好的,生成词云需要先安装`wordcloud`和`jieba`这两个库。可以通过以下命令安装:
```
pip install wordcloud
pip install jieba
```
然后,你需要准备一个文本文件,比如我们来使用《红楼梦》的文本作为例子,文件名为`hongloumeng.txt`。
接下来,可以使用以下代码生成词云:
```python
import jieba
from wordcloud import WordCloud
from PIL import Image
import numpy as np
# 打开文件并读取内容
with open('hongloumeng.txt', 'r', encoding='utf-8') as f:
text = f.read()
# 使用 jieba 分词
words = jieba.cut(text)
# 将分词结果转换成字符串
result = ' '.join(words)
# 加载背景图片
mask = np.array(Image.open("cloud.png"))
# 生成词云
wc = WordCloud(background_color="white", max_words=200, mask=mask, font_path='msyh.ttc')
wc.generate(result)
# 保存词云图片
wc.to_file("wordcloud.png")
```
上述代码中,我们首先使用`jieba`库对文本进行分词,然后将分词结果转换成字符串格式。接着,我们使用`WordCloud`类来生成词云,其中`background_color`表示背景颜色,`max_words`表示最多显示的词语数量,`mask`表示词云的形状,`font_path`表示字体路径。最后,我们将生成的词云图片保存到本地。注意,如果想要生成词云的形状,需要先准备好对应的图片。
阅读全文