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

main
phmycqkit 2 months ago
parent e2d32fa6d3
commit 777b21aa2f

@ -0,0 +1,23 @@
import pandas as pd
def GetAI():
# 读取CSV文件
df = pd.read_csv('danmu.csv')
# 筛选出第一列(弹幕)包含关键词"AI"的行
ai_danmu = df[df['弹幕'].str.contains('AI', na=False)]
# 将筛选后的数据保存为XLSX文件
ai_danmu.to_excel('ai_danmu.xlsx', index=False)
# 统计每个弹幕出现的次数
ai_danmu_count = ai_danmu.groupby('弹幕')['数量'].sum().reset_index()
# 找出出现次数最多的8个弹幕
ai_danmu_most_common = ai_danmu_count.sort_values(by='数量', ascending=False).head(8)
# 将这8个元素保存为XLSX文件
ai_danmu_most_common.to_excel('ai_danmu_most_common.xlsx', index=False)
if __name__ == '__main__':
GetAI()
Loading…
Cancel
Save