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

 

https://school.programmers.co.kr/learn/courses/30/lessons/250121

프로그래머스

문제풀이

SQL 조건, 정렬문을 알고리즘으로 구현하게끔 하는 문제다.

from functools import cmp_to_key


def solution(data, ext, val_ext, sort_by):
    cols = {"code": 0, "date": 1, "maximum": 2, "remain": 3}

    def cmp(x, y):
        return x[cols[sort_by]] - y[cols[sort_by]]

    # data에서 ext 값이 val_ext보다 이하인 데이터만 뽑음
    data = [row for row in data if row[cols[ext]] <= val_ext]

    # sort_by에 해당하는 값을 기준으로 오름차순으로 정렬
    answer = sorted(data, key=cmp_to_key(cmp))
    return answer