Delete 'word_cloud.py'

word_cloud.py

@@ -1,55 +0,0 @@
-"""
-    说明：写入execl，生成词云图
-"""
-# import pandas as pd
-from wordcloud import WordCloud
-import matplotlib.pyplot as plt
-from collections import Counter
-import openpyxl
-import jieba
-import re
-from bullet_screen import get_bullet_screen
-from bv_maker import get_bv, BV_NUM
-jieba.setLogLevel(jieba.logging.INFO)
-
-def confirm(bullet):
-    keywords = ['人工智能', 'AI', '机器学习', '深度学习', '神经网络', '自动驾驶', '自然语言处理', '智能', 'ai']  # 设置AI应用关键词
-    bullet = bullet.lower()
-    obj = r'(?<![a-zA-Z])(?:' + '|'.join(re.escape(keyword) for keyword in keywords) + r')(?![a-zA-Z])'
-    if re.search(obj, bullet):
-        return True
-    else:
-        return False
-
-def make_word_cloud(bullet_screen_list):
-    ai_related_bullet = []
-    for bullet in bullet_screen_list:
-        if confirm(bullet):  # 忽略大小写
-            ai_related_bullet.append(bullet)
-    # print(ai_related_bullet)
-    top8 =Counter(ai_related_bullet).most_common(8)
-    wb = openpyxl.Workbook()
-    sheet = wb.active
-    sheet.title = "AI Related Danmakus"
-    sheet.append(["弹幕内容", "出现次数"])
-    for item in top8:
-        sheet.append([item[0], item[1]])
-    wb.save('ai_bullet_screen.xlsx')
-    text = ' '.join(ai_related_bullet)
-    cut_text = ' '.join(jieba.cut(text))  # 使用jieba分词
-    wordcloud = WordCloud(
-        font_path='msyh.ttc',
-        width=800,
-        height=400,
-        background_color='white',
-        # colormap='cool'
-    ).generate(cut_text)
-    plt.figure(figsize=(10, 5))
-    plt.imshow(wordcloud, interpolation='bilinear')
-    plt.axis('off')
-    plt.show()  #展示词云图
-
-if __name__ == '__main__':
-    bv_list = get_bv(BV_NUM)
-    bullet_screen_list = get_bullet_screen(bv_list)
-    make_word_cloud(bullet_screen_list)