欢迎访问宙启技术站
智能推送

使用Python生成中文词云图

发布时间:2023-12-18 12:51:19

生成中文词云图可以使用jieba分词库和wordcloud词云库。下面是一个使用例子,首先需要安装jiebawordcloud库:

pip install jieba
pip install wordcloud

然后,使用以下代码生成中文词云图:

import jieba
from wordcloud import WordCloud
import matplotlib.pyplot as plt

# 读取文本数据
with open('example.txt', 'r', encoding='utf-8') as file:
    text = file.read()

# 使用jieba分词
words = ' '.join(jieba.cut(text))

# 生成词云图
wordcloud = WordCloud(font_path='simhei.ttf', width=800, height=400).generate(words)

# 显示词云图
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis('off')
plt.show()

在上面的代码中,假设要生成词云图的文本数据保存在example.txt文件中,通过jieba库进行中文分词,然后使用WordCloud类生成词云图。font_path参数指定需要使用的中文字体文件(这里假设使用simhei.ttf),widthheight参数指定词云图的宽度和高度。

通过运行以上代码,即可生成中文词云图,并显示出来。词云图的显示结果会根据词频进行布局,词频越高的词语在词云图中显示得越大。