文本处理+词云

Posted 2022-09-03 liveact

要求：使用本地文本内容，生成一张自定义的词云图

代码：

# 导入库
import wordcloud
import jieba
import imageio

# 打开文件
with open(r'D:\\Program Files\\Python\\Learn Python\\Python课程\\day04\\出师表.txt',encoding='gbk') as f:

    data = f.read()    #将内容存入data

data_list = jieba.lcut(data)    #将data中的数据用jieba库切开保存为列表
data = ' '.join(data_list)      #将列表中的元素用空字符拼接成字符串

# 找到一张图片
img = imageio.imread(r'D:\\Program Files\\Python\\Learn Python\\Python课程\\day05\\文本处理\\词云底图.jpg')

# 设置词云的字体，背景色，背景图
w = wordcloud.WordCloud(font_path=r'C:\\Windows\\Fonts\\simfang.ttf',background_color='white',mask=img)

# 生成词云
w.generate(data)

#保存为文件
w.to_file('出师表.png')