[Python] 10번 / 데이터 분석

✏️ 문제

 

문제 파악

중요한 건 sort key!!!

lambda 활용법을 더 익히자 

 

 

코드

def solution(data, ext, val_ext, sort_by):
    answer = []
    by = ["code", "date", "maximum", "remain"]

    for d in data:
        if d[by.index(ext)] < val_ext:
            answer.append(d)

    return sorted(answer, key=lambda x: x[by.index(sort_by)])