ADD file via upload

弹幕获取.py

@@ -0,0 +1,76 @@
+import requests
+from bs4 import BeautifulSoup
+import pandas as pd
+import re
+from wordcloud import WordCloud
+import matplotlib.pyplot as plt
+from openpyxl import Workbook
+
+def fetch_videos(keyword):
+    """
+    根据关键词搜索视频，并返回前300个视频的ID列表。
+    :param keyword: 搜索关键词
+    :return: 视频ID列表
+    """
+    url = "https://search.bilibili.com/all"
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3'
+    }
+    params = {
+        'keyword': keyword,
+        'order': 'totalrank',
+        'page': '1'
+    }
+    try:
+        response = requests.get(url, headers=headers, params=params)
+        response.raise_for_status()  # 抛出HTTP错误
+        soup = BeautifulSoup(response.text, 'html.parser')
+        video_ids = [a['href'].split('/')[-1] for a in soup.select('.video-item .title') if a.has_attr('href')]
+        print(f"Fetched {len(video_ids)} video IDs.")
+        return video_ids[:300]
+    except Exception as e:
+        print(f"Error fetching videos: {e}")
+        return []
+
+def fetch_danmaku(video_id):
+    """
+    获取指定视频的弹幕数据。
+    :param video_id: 视频ID
+    :return: 弹幕列表
+    """
+    url = f"https://api.bilibili.com/x/v2/dm/web/seg.so?type=1&oid={video_id}"
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3',
+        'Cookie': 'buvid3=0C047DB7-FB67-6565-B853-68B19196AEE053166infoc; buvid4=D2E32722-EB31-8B5B-8BC7-420F049CDE3657801-022071821-mG8+jYWtWHQ35A9yqIgZIA%3D%3D; buvid_fp=60e37bdf4fe67cde89d283db25adff46; _uuid=FCEA6C48-BB82-123A-61106-3F5410106BB410B03170infoc; b_nut=100; header_theme_version=CLOSE; enable_web_push=DISABLE; bmg_af_switch=1; bmg_src_def_domain=i0.hdslb.com; bsource=search_bing; bili_ticket=eyJhbGciOiJIUzI1NiIsImtpZCI6InMwMyIsInR5cCI6IkpXVCJ9.eyJleHAiOjE3MjY5MTczNjUsImlhdCI6MTcyNjY1ODEwNSwicGx0IjotMX0.Aa-1_tfEk0rFqyzRFZ-vIsSUSbvUfyR7woQA3IC3h0s; bili_ticket_expires=1726917305; CURRENT_FNVAL=4048; SESSDATA=aa6a6590%2C1742210524%2C7b1c4%2A92CjCxud8rqp6tuF7AYkzmJF0YS7_L4_80iMI3NuY5q-M7BEW3cf0_bVyhIcnZMJapP7YSVnJiQ2NVcTJZZ1ZIMFduRURJXzZXOWtaTTl2WnBFSHkwckM0UzdwY2xHMG9MNVl4c1pUSHlFaFJ4RnQ5WjY3ZHRtcm5qcDhNSVo3eXZORDczc0VlYlF3IIEC; bili_jct=5232d057d308c18c1419d19271a3b85e; DedeUserID=1576579979; DedeUserID__ckMd5=da7d6054e70acbba; b_lsid=674C255B_19204FCE528; sid=g83mkv7a; home_feed_column=4; browser_resolution=435-748',  # Cookie值
+        'Referer': f'https://www.bilibili.com/video/{video_id}',
+    }
+    try:
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()  # 抛出HTTP错误
+        data = response.json()
+        danmakus = [re.sub(r'<[^>]+>', '', dm['content']) for dm in data['data']['dm_seg_list']]
+        print(f"Fetched {len(danmakus)} danmakus for video {video_id}.")
+        return danmakus
+    except requests.RequestException as e:
+        print(f"Error fetching danmakus for video {video_id}: {e}")
+        return []
+    except KeyError as e:
+        print(f"Invalid response format for video {video_id}: {e}")
+        return []
+
+def analyze_danmakus(danmakus):
+    """
+    分析弹幕数据，统计与AI技术应用相关的弹幕数量。
+    :param danmakus: 弹幕列表
+    :return: 弹幕数量的字典
+    """
+    ai_keywords = ['AI', '人工智能', '机器学习', '深度学习', '算法', '大数据', '智能', '技术']
+    keyword_counts = {}
+    for danmaku in danmakus:
+        for keyword in ai_keywords:
+            if keyword in danmaku:
+                if danmaku in keyword_counts:
+                    keyword_counts[danmaku] += 1
+                else:
+                    keyword_counts[danmaku] = 1
+    return keyword_counts