Python酷库之旅-比翼双飞情侣库(04)
在Python中,我们可以使用jieba
库进行中文分词,并使用wordcloud
库生成词云。以下是一个简单的例子,展示如何使用这两个库来生成一个词云图。
首先,确保安装了所需的库:
pip install jieba wordcloud
然后,使用以下代码生成词云:
import jieba
from wordcloud import WordCloud
import numpy as np
import PIL.Image as Image
# 分词
text = "你好 世界 你好世界 你好世界 你好世界 你好世界 你好世界 你好世界"
cut_text = " ".join(jieba.cut(text))
# 创建词云
mask_image = np.array(Image.open("mask.png")) # 加载背景图片
wordcloud = WordCloud(background_color="white", mask=mask_image, font_path='simhei.ttf')
wordcloud.generate(cut_text)
# 保存词云图片
wordcloud.to_file("wordcloud.png")
在这个例子中,我们使用了一个简单的文本,并用jieba进行了分词。然后,我们使用一个图片作为词云的形状,并指定了一个字体文件(如simhei.ttf
),以生成带有宋体字的词云。最后,我们将生成的词云保存为一个PNG图片。
注意:需要一个合适的背景图片作为词云的“遮罩”,这里命名为mask.png
。同时,需要指定一个支持中文的字体文件,这里使用的是宋体(simhei.ttf
),你需要确保这个文件在你的文件系统中是可用的。如果你的环境中没有这个字体文件,可以从网上下载或者使用系统中已有的中文字体文件。
评论已关闭