[level 1] [PCCE 기출문제] 10번 / 데이터 분석 - 250121
성능 요약
메모리: 10.3 MB, 시간: 0.13 ms
구분
코딩테스트 연습 > PCCE 기출문제
채점결과
정확성: 100.0
합계: 100.0 / 100.0
제출 일자
2024년 1월 4일 12:5:35
문제 설명
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 함수를 완성해 주세요. 단, 조건을 만족하는 데이터는 항상 한 개 이상 존재합니다.
제한사항
- 1 ≤
data
의 길이 ≤ 500data[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
에 따라 올바른 범위의 숫자로 주어집니다.- 정렬 기준에 해당하는 값이 서로 같은 경우는 없습니다.
입출력 예
data | ext | val_ext | sort_by | result |
---|---|---|---|---|
[[1, 20300104, 100, 80], [2, 20300804, 847, 37], [3, 20300401, 10, 8]] | "date" | 20300501 | "remain" | [[3,20300401,10,8],[1,20300104,100,80]] |
입출력 예 설명
입출력 예 #1
- 본문의 내용과 동일합니다.
- cpp를 응시하는 경우 리스트는 배열과 동일한 의미이니 풀이에 참고해주세요.
- ex) 번호가 담긴 정수 리스트
numbers
가 주어집니다. => 번호가 담긴 정수 배열numbers
가 주어집니다.
- ex) 번호가 담긴 정수 리스트
- java를 응시하는 경우 리스트는 배열, 함수는 메소드와 동일한 의미이니 풀이에 참고해주세요.
- ex) solution 함수가 올바르게 작동하도록 한 줄을 수정해 주세요. => solution 메소드가 올바르게 작동하도록 한 줄을 수정해 주세요.
풀이
def solution(data, ext, val_ext, sort_by):
type = ['code', 'date', 'maximum', 'remain']
index_1 = type.index(ext)
index_2 = type.index(sort_by)
answer = []
for idx, d in enumerate(data):
if d[index_1] < val_ext:
answer.append(d)
answer = sorted(answer, key=lambda x: x[index_2])
return answer
느낀점
처음엔 pandas를 사용하여 데이터프레임으로 풀려고 시도했으나, 값을 넣는 과정과 정리하는 과정까지 더 번거로운점이 있어서 for loop을 사용하여 문제를 풀었다. index두개를 선언하고 ext와 sort_by를 기억하고 for loop을 돌면서 val_ext보다 작은 값은 answer에 append하고 마지막에 index_2를 기준으로 오름차순 정리를 하면 정답이 나오게 된다.
프로그래머스 level1을 기준으로 6문제 정도가 남았는데 얼른 다 끝내고 level2로 들어가면 좋겠다ㅏㅏㅏ
출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges
'알고리즘' 카테고리의 다른 글
프로그래머스: 개인정보 수집 유효기간 (0) | 2024.01.23 |
---|---|
프로그래머스: 바탕화면 정리 (0) | 2024.01.19 |
프로그래머스: 성격 유형 검사하기 (1) | 2024.01.18 |
프로그래머스: 신규 아이디 추천 (1) | 2024.01.17 |
프로그래머스: 햄버거 만들기 (0) | 2024.01.17 |