使用Ai做词云图
时间: 2023-06-28 22:05:43 浏览: 80
做词云图的基本思路是先对文本进行分词,然后统计每个词出现的频率,最后将词频高的词汇以不同的字体大小和颜色展示在一张图片上。
下面是使用Python中的wordcloud库来实现词云图的例子:
```python
import jieba
from wordcloud import WordCloud
# 读入文本文件
with open('input.txt', 'r', encoding='utf-8') as f:
text = f.read()
# 对文本进行分词
words = jieba.cut(text)
# 统计词频
word_dict = {}
for word in words:
if word not in word_dict:
word_dict[word] = 0
word_dict[word] += 1
# 生成词云图
wc = WordCloud(background_color='white', font_path='simhei.ttf')
wc.generate_from_frequencies(word_dict)
wc.to_file('output.png')
```
其中,`jieba`库用于对中文文本进行分词,`WordCloud`库则用于生成词云图。需要注意的是,在生成词云图之前需要安装`wordcloud`和`jieba`库,并且下载一个中文字体文件(在本例中使用的是`simhei.ttf`)。
这段代码会将`input.txt`文件中的文本进行分词和词频统计,最后生成一张词云图并保存为`output.png`文件。你可以根据自己的需求修改代码中的输入文件名、输出文件名等参数。