[Programmers] [PCCE 기출문제] 10번 / 데이터 분석 - 250121

lambda를 이용해서 정렬하기 !

 

Question

 

정렬한 데이터들이 담긴 이차원 정수 리스트 data와 어떤 정보를 기준으로 데이터를 뽑아낼지를 의미하는 문자열 ext, 뽑아낼 정보의 기준값을 나타내는 정수 val_ext, 정보를 정렬할 기준이 되는 문자열 sort_by가 주어집니다.

data에서 ext 값이 val_ext보다 작은 데이터만 뽑은 후, sort_by에 해당하는 값을 기준으로 오름차순으로 정렬하여 return 하도록 solution 함수를 완성해 주세요. 단, 조건을 만족하는 데이터는 항상 한 개 이상 존재합니다.

 

 

Code

answer이 2차원 배열이기 때문에 x는 각 요소(1차원 배열)이고, 그 x의 criteria[sort_by] (==index 번호) 를 기준으로 정렬.

def solution(data, ext, val_ext, sort_by):
    answer = []
    criteria = {'code':0,'date':1,'maximum':2,'remain':3}
    
    for d in data:
        if d[criteria[ext]] < val_ext:
            answer.append(d)
            
    return sorted(answer, key = lambda x:x[criteria[sort_by]])

 

now me

On my github