From f36ccff02f20ee69f939221cad615b1df4332b85 Mon Sep 17 00:00:00 2001 From: Yener Date: Wed, 27 Nov 2019 10:28:20 +0800 Subject: [PATCH] fix range --- jiagu/textrank.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/jiagu/textrank.py b/jiagu/textrank.py index ad8ef68..1e469e1 100644 --- a/jiagu/textrank.py +++ b/jiagu/textrank.py @@ -78,11 +78,8 @@ class Keywords(object): tol=self.__tol) sent_selected = nlargest(n, zip(scores, count())) sent_index = [] - for i in range(n): - try: - sent_index.append(sent_selected[i][1]) - except: - pass + for i in range(min(len(sent_selected), n)): + sent_index.append(sent_selected[i][1]) return [index_word[i] for i in sent_index]