wordcloud分词生成

友猫2024-10-22 9:57

代码如下 _

复制代码

from wordcloud import WordCloud
import PIL.Image as image
import numpy as np
import jieba

def cut(text):
    word_list = jieba.cut(text,cut_all= True)
    # 分词后在单独个体之间加上空格
    result = " ".join(word_list)
    return result

#导入文本文件,进行分词,制作词云
with open("xiaoshuo.txt") as fp:
    text = fp.read()
    # 将读取的中文文档进行分词
    text = cut(text)
    
#设置词云形状
mask = np.array(image.open("538eba130bea1e8c.webp"))
#自定义词云
wordcloud = WordCloud(mask=mask,background_color='#FFFFFF',font_path=r'C:\windows\Fonts\汉仪书魂体简.TTF').generate(text)

#返回对象
image_produce = wordcloud.to_image()
#保存图片
wordcloud.to_file("new_wordcloud.jpg")
#显示图像
image_produce.show()

使用的文本 _

1.xiaoshuo.txt文件

2.背景为白色的一张图片，538eba130bea1e8c.webp

结果