https://programmers.co.kr/learn/courses/30/lessons/42579?language=python3

best_album.py

from collections import defaultdict

def solution(genres, plays):
    answer = []
    new_genres = defaultdict(int)
    new_play = defaultdict(list)
    for index,(k,v) in enumerate(zip(genres,plays)):
        new_genres[k] += v
        new_play[k].append((index,-v))
    new_order = sorted(new_genres.keys(),key=lambda k : new_genres[k],reverse=True)
    for i in new_order:
        answer.extend([pk for pk,value in sorted(new_play[i],key=lambda t : t[1])[:2]])

    return answer