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

 

📜 문제 설명

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

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

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

 

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

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

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

 

제한사항

• 1 ≤ data의 길이 ≤ 500
  • data[i]의 원소는 [코드 번호(code), 제조일(date), 최대 수량(maximum), 현재 수량(remain)] 형태입니다.
  • 1 ≤ 코드 번호≤ 100,000
  • 20000101 ≤ 제조일≤ 29991231
  • data[i][1]은 yyyymmdd 형태의 값을 가지며, 올바른 날짜만 주어집니다. (yyyy : 연도, mm : 월, dd : 일)
  • 1 ≤ 최대 수량≤ 10,000
  • 1 ≤ 현재 수량≤ 최대 수량
• ext와 sort_by의 값은 다음 중 한 가지를 가집니다.
  • "code", "date", "maximum", "remain"
  • 순서대로 코드 번호, 제조일, 최대 수량, 현재 수량을 의미합니다.
• val_ext는 ext에 따라 올바른 범위의 숫자로 주어집니다.
• 정렬 기준에 해당하는 값이 서로 같은 경우는 없습니다.

 

🔗 문제 링크

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

프로그래머스

 

💡 문제 풀이

오랜만에 올리는 문제 풀이( ˃ ⩌˂)

단순하게 접근할 수 있는 문제다. 파라미터가 좀 많아서 헷갈리긴 했다. (중간에 파라미터 틀려서 와르르 틀렸음)

우선 조건에 맞는 데이터들만 남겨 놓는다. 이 때 조건의 기준이 뭐가 될지 몰라서 조건이 될 수 있는 요소들의 목록을 만든 후 index 메서드를 이용해 조건을 찾아낸다. 

그 후에 람다 함수를 사용해 정렬하면 끝.

def solution(data, ext, val_ext, sort_by):
    val = ['code', 'date', 'maximum', 'remain']
    idx = val.index(ext)
    
    answer = [row for row in data if row[idx] <= val_ext]
            
    answer = sorted(answer, key = lambda x :x[val.index(sort_by)])
    return answer