You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.3 KiB
41 lines
1.3 KiB
2 months ago
|
|
||
|
|
|
||
|
|
import os
|
||
|
|
from os import path
|
||
|
|
from wordcloud import WordCloud
|
||
|
|
import jieba
|
||
|
|
from imageio.v3 import imread
|
||
|
|
|
||
|
|
def Bulletchat_Wordcloud():
|
||
|
|
try:
|
||
|
|
#获取当前文件路径
|
||
|
|
d = path.dirname(__file__) if "__file__" in locals() else os.getcwd()
|
||
|
|
text = open(path.join(d,r'2028.txt'), 'rb').read()
|
||
|
|
# 设置模板图
|
||
|
|
img_mask = imread(r'm1.png')
|
||
|
|
# 对弹幕进行精确模式分词
|
||
|
|
text_list = jieba.lcut(text,cut_all=False)
|
||
|
|
text_str = ' '.join(text_list)
|
||
|
|
# print(text_str)
|
||
|
|
# 设置中文字体
|
||
|
|
font_path = r'C:\Users\15653\msyh.ttc'
|
||
|
|
# 停止词设置
|
||
|
|
stopwords = set('')
|
||
|
|
stopwords.update(['的','和','又','了','都','是','什么','所以','这','呢','吧','吗','个','呀','嘛','哈'])
|
||
|
|
wc = WordCloud(
|
||
|
|
font_path=font_path,
|
||
|
|
#max_words=500, # 最多词个数
|
||
|
|
#min_font_size=15,
|
||
|
|
#max_font_size=100,
|
||
|
|
mask=img_mask,
|
||
|
|
background_color='white',
|
||
|
|
stopwords=stopwords,
|
||
|
|
colormap='copper'
|
||
|
|
)
|
||
|
|
wc.generate(text_str)
|
||
|
|
wc.to_file(r'outm1.png')
|
||
|
|
except Exception as e:
|
||
|
|
print(f"词云图生成异常: {e}")
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
Bulletchat_Wordcloud()
|