Pythonで指定されたリストからk個の最長の単語を検索する

さまざまな長さの多くの単語を含むリストから、上位n個の最長の単語を選択する必要があるシナリオがあります。この記事では、それを実現するためのさまざまなアプローチを紹介します。

count（）およびsorted（）を使用

まず、リストの要素を逆の順序で並べ替えて、リストの先頭で最も長い単語が使用できるようにします。次に、各単語の長さを見つけて、カウントの結果を変数に追加します。最後に、必要な最長の単語の数を切り取ります。

例

from itertools import count

def longwords(l, x):
   c = count()
   return sorted(l, key=lambda i: (len(i), next(c)),
                  reverse=True)[:x]

listA = ['Earth','Moonshine','Aurora','Snowflakes','Sunshine']
n = 2
print(longwords(listA, n))

出力

上記のコードを実行すると、次の結果が得られます-

['Snowflakes', 'Moonshine']

zipと列挙を使用

このアプローチでは、enumerateを使用してリストの各要素をリストし、sortedおよびzip関数を適用してカウントを取得します。負の長さの値は、並べ替えの逆の順序を示し、最後に必要な数のカウントをスライスします。

例

def longwords(l, x):
   idx, words = zip(*sorted(enumerate(l),
   key = lambda i: (-len(i[1]), -i[0]))[:x])
   return list(words)

listA = ['Earth','Moonshine','Aurora','Snowflakes','Sunshine']
n = 2
print(longwords(listA, n))

出力

上記のコードを実行すると、次の結果が得られます-

['Snowflakes', 'Moonshine']