统计AI技术应用方面的每种弹幕数量,并输出数量排名前8的弹幕

main
phmycqkit 2 months ago
parent 6461d2c8c7
commit ec0a20ec5f

@ -0,0 +1,24 @@
import pandas as pd
from collections import Counter
def GetAI():
# 读取弹幕文件
with open('danmu.csv', 'r', encoding='utf-8-sig') as f:
danmus = f.readlines()
# 筛选与AI技术应用相关的弹幕
keywords = ['ai','Ai','AI']
ai_danmu = [danmu for danmu in danmus if any(keyword in danmu for keyword in keywords)]
# 统计弹幕数量
ai_danmu_counter = Counter(ai_danmu)
ai_danmu_most_common = ai_danmu_counter.most_common(8)
df = pd.DataFrame(ai_danmu_most_common, columns=['弹幕', '数量'])
# 保存为Excel文件
filename = 'ai_danmu_most_common.xlsx'
df.to_excel(filename, index=False)
if __name__ == '__main__':
GetAI()
Loading…
Cancel
Save