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

---

문제설명

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

예를 들어 다음과 같이 데이터가 주어진다면,

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

이 데이터는 다음 표처럼 나타낼 수 있습니다.

code	date	maximum	remain
1	20300104	100	80
2	20300804	847	37
3	20300401	10	8

주어진 데이터 중 "제조일이 20300501 이전인 물건들을 현재 수량이 적은 순서"로 정렬해야 한다면 조건에 맞게 가공된 데이터는 다음과 같습니다.

data = [[3,20300401,10,8],[1,20300104,100,80]]

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

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

 

제한사항

 

입출력 예

 

---

문제풀이

• 해당문제는 설명은 길게 많으나 지시사항들을 하나씩 따라가다 보면 손쉽게 풀 수 있는 문제였다.
• 이번 문제를 통해서 파이썬에서 제공해주는 강력한 기능 중 하나일 sort함수를 제대로 이해하면 좋을 것 같아 아래 잘 정리된 블로그를 추가해본다.( https://haesoo9410.tistory.com/193 )
• 그리고 마지막에 이상하게 return answer.sort(key = lambda x : x[dict[sort_by]]) 로 마무리를 하면 null을 반환하는 에러가 있었다. 왜 그런지 sort함수가 파이썬 내부적으로 구현된 방식 때문인것 같은데... 그 이유를 확실히 찾아내지는 못했다.. 혹시 아시는 분이 계시다면 댓글로 지식을 나눔부탁드립니다.

코드 & 설명

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

 

 

<파이썬> 2차원 배열 정렬