# 运行程序前，请务必前往本课包【课程说明】处，下载课程素材，完成配置
# 编程前，运行【安装第三方库.py】文件，安装本课用到的第三方库

import jieba
import wordcloud
from PIL import Image
import numpy as np

# 加载背景图片
mask_image = np.array(Image.open("心.png"))

with open('春.txt', 'r', encoding='utf-8') as f:
    text = f.read()

lst = jieba.lcut(text)
m = ' '.join(lst)

print('开始制作词云...')

w = wordcloud.WordCloud(
    font_path = '大梁体繁简.TTF',
    stopwords = ['着',"了","里",'的'],
    # 使用参数background_color设置图片背景
    background_color = None,
    mode = 'RGBA',
    mask = mask_image
)

w.generate(m)
w.to_file('词云-春.png')

print('词云图片已生成,开始展示图片')

p = Image.open('词云-春.png')
p.show()
print('图片展示完成')