|
|
|
|
import jieba
|
|
|
|
|
from matplotlib import pyplot as plt
|
|
|
|
|
from wordcloud import WordCloud
|
|
|
|
|
from PIL import Image
|
|
|
|
|
import numpy as np
|
|
|
|
|
|
|
|
|
|
def predeal(txt):
|
|
|
|
|
cut = jieba.cut(txt) # 分词
|
|
|
|
|
string = ' '.join(cut)
|
|
|
|
|
return string
|
|
|
|
|
|
|
|
|
|
def drawcloud(imgpath, stopword, string, savepath):
|
|
|
|
|
try:
|
|
|
|
|
img = Image.open(imgpath) # 打开图片
|
|
|
|
|
except IOError as e:
|
|
|
|
|
print(f"打开图片时出错：{e}")
|
|
|
|
|
return
|
|
|
|
|
img_array = np.array(img) # 将图片转换为数组
|
|
|
|
|
wordcloud = WordCloud(
|
|
|
|
|
background_color='white',
|
|
|
|
|
width=1000,
|
|
|
|
|
height=800,
|
|
|
|
|
mask=img_array, # 设置背景图片
|
|
|
|
|
font_path='C:\\Windows\\Fonts\\msyh.ttc', # 电脑自带的字体
|
|
|
|
|
stopwords=stopword,
|
|
|
|
|
colormap='plasma'
|
|
|
|
|
)
|
|
|
|
|
wordcloud.generate_from_text(string) # 绘制图片
|
|
|
|
|
plt.imshow(wordcloud)
|
|
|
|
|
plt.axis('off')
|
|
|
|
|
wordcloud.to_file(savepath) # 保存图片
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
txtpath = r'd:\学习\软件工程\swork\res\select_ai.txt'
|
|
|
|
|
imgpath = r'd:\学习\软件工程\swork\tu4.png'
|
|
|
|
|
savepath = r'd:\学习\软件工程\swork\cloud_ai.png'
|
|
|
|
|
stopword = ['啊', '的', '是', '视频', '个', '哈', '第', '哈哈哈', '吧', '了', '我', '你','福州大学'] # 不想显示的词
|
|
|
|
|
try:
|
|
|
|
|
with open(txtpath, 'r', encoding='utf-8') as file:
|
|
|
|
|
txt = file.read()
|
|
|
|
|
except FileNotFoundError as e:
|
|
|
|
|
print(f"读取文本文件时出错：{e}")
|
|
|
|
|
else:
|
|
|
|
|
string = predeal(txt)
|
|
|
|
|
drawcloud(imgpath, stopword, string, savepath)
|
