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

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

https://school.programmers.co.kr/learn/courses/30/lessons/250121?language=python3

문제 설명

데이터를 분석하는 작업을 진행하고 있습니다.

데이터는 ["코드 번호(code)", "제조일(date)", "최대 수량(maximum)", "현재 수량(remain)"]으로 구성되어 있으며 현식이는 이 데이터들 중 조건을 만족하는 데이터만 뽑아서 정렬하려 합니다.

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

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

풀이 과정

1. 데이터의 코드 번호, 제조일, 최대 수량, 현재 수량을 접근하기 위한 인덱스 dict를 생성
2. 각 데이터에서 val_ext보다 작거나 같은 값을 list로 생성
3. 생성된 list를 sort_by를 기준으로 정렬

def solution(data, ext, val_ext, sort_by):
    
    dic = { "code":0, "date":1, "maximun":2, "remain":3, }
    
    answer = [d for d in data if d[dic[ext]] <= val_ext]

    r = sorted(answer, key = lambda x : x[dic[sort_by]])
    return r