[프로그래머스] 250121 PCCE 기출문제 10번 / 데이터 분석 파이썬

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

프로그래머스

 

1. 문제 설명

data = [[1, 20300104, 100, 80], [2, 20300804, 847, 37], [3, 20300401, 10, 8]]

이런 식으로 구성된 데이터에서 원하는 정보를 뽑아 정렬해 출력하는 문제다.

 

 

2. 문제 풀이

def solution(data, ext, val_ext, sort_by):
    answer = []
    data_type = {
        "code" : 0,
        "date" : 1,
        "maximum" : 2,
        "remain" : 3
        }
    for i in data:
        if i[data_type[ext]] < val_ext:
            answer.append(i)
            
    answer.sort(key=lambda x: x[data_type[sort_by]])
    
    return answer

 

처음에는 조건문을 쓰는 방법을 떠올렸지만, 정렬할 때 또 조건문을 쓰는 건 너무 길어질 것 같아 data_type 딕셔너리를 만들어두었다.

 

val_ext보다 작은 리스트들을 넣어놓고, 마지막에 sort_by에 맞는 리스트의 인덱스 값들을 비교해 정렬해주었다.