リストのソート後の上位5つのみリストに格納する方法がわかりません

Question

res =
[{'search_count':4, 'trend': 'WORDA'}, {'search_count': 3, 'trend': 'WORDB'}, {'search_count': 2, 'trend': 'WORDBXXX'}, {'search_count': 2, 'trend': 'XXWORDA'}, {'search_count': 2, 'trend': 'XXXWORDB'}, {'search_count': 1, 'trend': 'あああいいい'}], {'search_count': 1, 'trend': 'あああえええ'}, {'search_count': 2, 'trend': 'あああううう'}, {'search_count': 2, 'trend': 'あああおおお'}

top5_trend = sorted(res, key=lambda x: x['search_count'], reverse=True)
# top5_trend =         
  [
   {'search_count': 4, 'trend': 'WORDA'}, 
   {'search_count': 3, 'trend': 'WORDB'}, 
   {'search_count': 2, 'trend': 'WORDBXXX'}, 
   {'search_count': 2, 'trend': 'XXWORDA'}, 
   {'search_count': 2, 'trend': 'XXXWORDB'}, 
   {'search_count': 2, 'trend': 'あああおおお'},
   {'search_count': 2, 'trend': 'あああううう'},
   {'search_count': 1, 'trend': 'あああいいい'},
   {'search_count': 1, 'trend': 'あああえええ'}
  ]

top5_trendのリストから'search_count'キーの数値が大きい上位5位のみをtop5_trendに格納したいと思ってます。
また、上位5番目と同じ'search_count'の値を持つものも一緒に格納したいと思っています。
つまり、今回の場合ですと、以下のようにしたいと考えています。

top5_trend =[
    {'search_count': 4, 'trend': 'WORDA'}, 
    {'search_count': 3, 'trend': 'WORDB'}, 
    {'search_count': 2, 'trend': 'WORDBXXX'}, 
    {'search_count': 2, 'trend': 'XXWORDA'}, 
    {'search_count': 2, 'trend': 'XXXWORDB'}, 
    {'search_count': 2, 'trend': 'あああいいい'},
    {'search_count': 2, 'trend': 'あああううう'}
]

実現するために下のようにコードを書いたののですが、うまく実現できません。
具体的には、
if top5_trend[i-1]:
を書いてるにも関わらず、
IndexError: list index out of range
のエラーが出力されてしまうことや、
4行目のfor文を回して、search_countが次の要素とかぶっていなければfor文を止めるやり方がわかっておらず、実現できていないのかと思っています。

よろしくお願いします。

if len(top5_trend) <= 5: #リスト内が5以下の場合
                ans = {"recommend_trend": top5_trend}
else: #リスト内が5以上の場合
    for i in range(0, len(top5_trend)):
        if i >= 5:
            if top5_trend[i-1]:
                if i<= len(top5_trend)-1:
                        if not top5_trend[i-1]["search_count"] == top5_trend[i]["search_count"]:
                            pop_val = len(top5_trend) - 5  
                            del top5_trend[i:]
                            print(top5_trend)
                        if  top5_trend[i-1]["search_count"] == top5_trend[i]["search_count"]: 
                            print("pass")
                            

Comments

[meg_]

提示のコード（最初のコード）がエラーで実行出来ません。

Answer 1

エラーが出る原因は、以下の文でリストの長さが減っているからです。

del top5_trend[i:]

以下のコードでやってください。

>>> res = [{'search_count': 4, 'trend': 'WORDA'},
...        {'search_count': 3, 'trend': 'WORDB'},
...        {'search_count': 2, 'trend': 'WORDBXXX'},
...        {'search_count': 2, 'trend': 'XXWORDA'},
...        {'search_count': 2, 'trend': 'XXXWORDB'},
...        {'search_count': 1, 'trend': 'あああいいい'},
...        {'search_count': 1, 'trend': 'あああえええ'},
...        {'search_count': 2, 'trend': 'あああううう'},
...        {'search_count': 2, 'trend': 'あああおおお'}]
>>>
>>> top5_trend = sorted(res, key=lambda x: x['search_count'], reverse=True)
>>>
>>> top5 = [x for x in top5_trend if x['search_count'] >= top5_trend[4]['search_count']]
>>> len(top5)
7
>>> for x in top5:
...     print(x)
...
{'search_count': 4, 'trend': 'WORDA'}
{'search_count': 3, 'trend': 'WORDB'}
{'search_count': 2, 'trend': 'WORDBXXX'}
{'search_count': 2, 'trend': 'XXWORDA'}
{'search_count': 2, 'trend': 'XXXWORDB'}
{'search_count': 2, 'trend': 'あああううう'}
{'search_count': 2, 'trend': 'あああおおお'}

ようするに、並べ替えたものの5番目(top5_trend[4])と比べて大きいか等しいものを取り出せば良いのです。

Comments

[1mzmk]

ご回答ありがとうございます。
ソートをすでにしているなら、五番目の要素と比べてれば良いという発想ができませんでした。とても勉強になりました。
ありがとうございました。

Answer 2

if top5_trend[i-1]:

を書いてるにも関わらず、
IndexError: list index out of range
のエラーが出力されてしまうことや、

エラーが発生しているのはif not top5_trend[i-1]["search_count"] == top5_trend[i]["search_count"]:のtop5_trend[i]の方ではありませんか？

---

ソート済のリストであれば下記のようなコードでも良いかと思います。

res = [{'search_count':4, 'trend': 'WORDA'}, {'search_count': 3, 'trend': 'WORDB'}, {'search_count': 2, 'trend': 'WORDBXXX'}, {'search_count': 2, 'trend': 'XXWORDA'},
       {'search_count': 2, 'trend': 'XXXWORDB'}, {'search_count': 1, 'trend': 'あああいいい'}, {'search_count': 1, 'trend': 'あああえええ'},
       {'search_count': 2, 'trend': 'あああううう'}, {'search_count': 2, 'trend': 'あああおおお'}]
top5_trend = sorted(res, key=lambda x: x['search_count'], reverse=True)

ans = []
num = -1
count = 0
for i in top5_trend:
    if num == i['search_count']:
        ans.append(i)
        count += 1
    else:
        if count < 5:
            num = i['search_count']
            ans.append(i)
            count += 1

Comments

[1mzmk]

ご回答ありがとうございました。