Update select_content.py

main
ptkbf2lr5 2 months ago
parent 99a7f3587d
commit be62c0356d

@ -5,10 +5,10 @@ from collections import Counter
with open('all_content.txt', mode='r', encoding='utf-8') as f: with open('all_content.txt', mode='r', encoding='utf-8') as f:
data_list = f.readlines() data_list = f.readlines()
# AI技术应用有关关键词 # AI技术应用有关关键词
keywords = ['AI','人工智能','ai音效','ai视频','ai技术','机器学习', '深度学习', '自然语言处理','ai训练', ai_keywords = ['AI','人工智能','ai音效','ai视频','ai技术','机器学习', '深度学习', '自然语言处理','ai训练',
'大模型','自然语言处理','云计算','神经网络', '自动驾驶','ai设计','ai图','ai软件',] '大模型','自然语言处理','云计算','神经网络', '自动驾驶','ai设计','ai图','AI软件',]
# 筛选有关AI的弹幕 # 筛选有关AI的弹幕
selectdanmu = [danmu for danmu in data_list if any(keyword in danmu for keyword in keywords)] selectdanmu = [danmu for danmu in data_list if any(keyword in danmu for keyword in ai_keywords)]
# 统计弹幕数量 # 统计弹幕数量
num = Counter(selectdanmu) num = Counter(selectdanmu)

Loading…
Cancel
Save