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

코딩테스트 연습>해시>베스트앨범

import numpy

def solution(genres, plays):
    answer = []
    
    music = list(zip(range(len(genres)),genres, plays))
    music = sorted(music, key=lambda x : (-x[2], x[0]))
    music = numpy.array(music)
    
    dic_music = {}
    for s in set_genres:
        dic_music[s] = 0
        mi = numpy.where(music == s)[0]
        for i in mi:
            dic_music[s] += int(music[i][2])
    
    dic_music = sorted(dic_music.items(), reverse=True, key=lambda item: item[1])
    
    for g, p in dic_music:
        mi = numpy.where(music == g)[0]
        if len(mi) == 0:
            answer.append(int(music[mi[0]][0]))
        else:
            for i in mi[:2]:
                answer.append(int(music[i][0]))        
        
    return answer