[Lv.1 / 프로그래머스 / 파이썬] 데이터 분석 (PCCE 기출문제 10번)

 

 

 

프로그래머스

 

 

풀이

 

리스트에 담겨 있는 자료를 조건에 따라 적절히 필터링하고, 정렬하여 출력하는 문제이다.

 

sort 함수의 key 파라미터로 람다 함수를 넘겨 보다 쉽게 정렬할 수 있다.

 

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