如何用python生成词云图而且该词云图能够更改蒙版请写出代码用中文回答
使用Python生成词云图需要使用第三方库wordcloud和matplotlib。首先确保已经安装这两个库。
接下来,我们可以使用以下代码来生成词云图并更改蒙版:
import jieba # 分词库
from wordcloud import WordCloud, STOPWORDS
import matplotlib.pyplot as plt
from PIL import Image
# 读取文本内容
with open("text.txt", "r", encoding="utf-8") as f:
text = f.read()
# 使用jieba进行中文分词
word_list = jieba.cut(text)
words = " ".join(word_list)
# 读取蒙版图片
mask = np.array(Image.open("mask.png"))
# 设置停用词
stopwords = set(STOPWORDS)
stopwords.add("的") # 添加自定义的停用词
# 创建词云对象
wc = WordCloud(background_color="white", mask=mask, stopwords=stopwords)
# 生成词云图
wc.generate(words)
# 显示词云图
plt.imshow(wc, interpolation="bilinear")
plt.axis("off")
plt.show()
# 保存词云图
wc.to_file("wordcloud.png")
在上述代码中,我们首先使用jieba库对文本进行中文分词,然后将分词结果拼接成一个字符串。接着,我们使用PIL库读取蒙版图片,并将其转换为numpy数组。然后,我们设置了停用词集合,可以根据需要添加自定义的停用词。接下来,我们创建了一个词云对象,设置了背景颜色、蒙版和停用词。最后,我们使用generate()方法生成词云图,并使用imshow()方法显示词云图。你还可以使用to_file()方法保存词云图。
请确保将代码中的text.txt替换为你的文本文件路径,mask.png替换为你的蒙版图片路径
原文地址: https://www.cveoy.top/t/topic/h5LA 著作权归作者所有。请勿转载和采集!