10. 데이터 분석 ★ (이차원 배열의 정렬 key)

def solution(data, ext, val_ext, sort_by):
    answer = []
    val = ["code", "date", "maximum", "remain"]
    excode = val.index(ext)
    sortcode = val.index(sort_by)
    for d in data:
        if d[excode] < val_ext:
            answer.append(d)
    answer.sort(key = lambda x:x[sortcode])
    return answer

 

유형 : 구현

풀이 시간: 약 30분

 

answer에 append까지는 쉬웠는데,

이차원배열 정렬에서 조금 애먹었다. 

익명함수 lambda를 이용하면 쉽게 정렬할 key를 지정할 수 있다. 

 

    answer.sort(key = lambda x:x[sortcode])