from wordcloud import WordCloud  
from PIL import Image  
import matplotlib.pyplot as plt  
import numpy as np
  
# 定义文本文件路径  
text_file_path = 'AI弹幕·生成词云图用.txt'  
# 定义遮罩图片路径（确保是具有Alpha通道的PNG图片）  
mask_image_path = 'MASK.png'  
  
# 读取文本文件  
with open(text_file_path, 'r', encoding='utf-8') as file:  
    txt = file.read()  
  
# 加载遮罩图片  
mask = np.array(Image.open(mask_image_path))  
  
# 注意：WordCloud 可以直接处理具有Alpha通道的PNG图片，所以这里不需要额外处理  
  
# 创建WordCloud对象，并设置mask  
wordcloud = WordCloud(  
    background_color="white",  
    width=800,  
    height=600,  
    max_words=200,  
    max_font_size=80,  
    mask=mask,  
    contour_width=3,  
    contour_color='steelblue',  
    font_path="msyh.otf"  # 确保字体文件路径正确  
).generate(txt)  
  
# 保存词云为图片文件  
wordcloud.to_file('词云图.png')