Make LevenshteinSimilarityIndex.most_similar easier to read

gensim/similarities/levenshtein.py

@@ -11,6 +11,7 @@
 import logging
 from math import floor
 
+from gensim.matutils import argsort
 from gensim.similarities.termsim import TermSimilarityIndex
 
 logger = logging.getLogger(__name__)
@@ -139,15 +140,10 @@ def __init__(self, dictionary, alpha=1.8, beta=5.0, threshold=0.0):
         super(LevenshteinSimilarityIndex, self).__init__()
 
     def most_similar(self, t1, topn=10):
-        similarities = (
-            (levsim(t1, t2, self.alpha, self.beta, self.threshold), t2)
-            for t2 in self.dictionary.values()
-            if t1 != t2
-        )
-        most_similar = (
-            (t2, similarity)
-            for (similarity, t2) in sorted(similarities, reverse=True)
-            if similarity > 0
-        )
-        for _, (t2, similarity) in zip(range(topn), most_similar):
-            yield (t2, similarity)
+        terms = [t2 for t2 in self.dictionary.values() if t1 != t2]
+        similarities = [levsim(t1, t2, self.alpha, self.beta, self.threshold) for t2 in terms]
+        for index in argsort(similarities, topn, reverse=True):
+            t2 = terms[index]
+            similarity = similarities[index]
+            if similarity > 0:
+                yield (t2, similarity)