[Python][프로그래머스][Level3] 베스트앨범

def solution(genres, plays):
    answer =[]
    
    music={genre: [0,[]] for genre in genres}
    for idx, genre in enumerate(genres):
        music[genre][0]+=plays[idx]
        music[genre][1].append(idx)

    res = sorted(music, key= lambda x : -music[x][0])
    for genre in res:
        answer +=sorted(music[genre][1], key = lambda x: -plays[x])[:2]
    return answer