parent
68973bc7e4
commit
bf5b351eda
@ -1,37 +0,0 @@
|
||||
from get_danmu import get_danmu
|
||||
#def get_danmu(urls):
|
||||
|
||||
|
||||
def is_related(danmu, keyword_string): #判断弹幕是否和keyword相关
|
||||
for string in keyword_string:
|
||||
if string in danmu:
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def danmu_about_AI(urls,keywords):
|
||||
all_danmu = get_danmu(urls)
|
||||
|
||||
#初始化保存结果的字典
|
||||
keyword_count = {i: 0 for i in keywords} #保存数量
|
||||
keyword_dist = {i: [] for i in keywords} #保存相关弹幕
|
||||
|
||||
|
||||
#遍历每一个弹幕,若和某一个关键字相关则统计
|
||||
for danmu in all_danmu:
|
||||
for keyword, keyword_string in keywords.items():
|
||||
if is_related(danmu, keyword_string):
|
||||
keyword_count[keyword] += 1
|
||||
keyword_dist[keyword].append(danmu)
|
||||
break
|
||||
else:
|
||||
continue
|
||||
#得到前八个keyword和他们的count
|
||||
top_8_AI = sorted(keyword_count.items(), key=lambda item: item[1], reverse=True)[:8]
|
||||
|
||||
#统计得到前八个相关技术的所有弹幕
|
||||
top_8_danmu = []
|
||||
for keyword,_ in top_8_AI:
|
||||
for danmu in keyword_dist[keyword]:
|
||||
top_8_danmu.append(danmu)
|
||||
return top_8_danmu,top_8_AI
|
Loading…
Reference in new issue