프로그래머스 - '데이터 분석' [코딩테스트 고득점 Kit/PCCE 기출문제 10번]

# 데이터 분석 # 250121번 # lv1

 

# 문제

---

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

data = [[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
[[1, 20300104, 100, 80], [2, 20300804, 847, 37], [3, 20300401, 10, 8]]

ext	
"date"	

val_ext
20300501

sort_by
"remain"	

result
[[3,20300401,10,8],[1,20300104,100,80]]

# 풀이

---

# 풀이1

def solution(data, ext, val_ext, sort_by):
    # ext = '어떤 정보를 기준으로 데이터 뽑을 지'
    # val_ext = '뽑아온 정보의 기준값을 나타내는 정수'
    # sort_by = '정보를 정렬할 기준'
    # data에서 ext를 추출하고 val_ext 보다 작으면 da 리스트를 tmp_list에 더함. 
    
    if ext == 'code':
    	tmp_list = [da for da in data if da[0] < val_ext]
    elif ext == 'date':
    	tmp_list = [da for da in data if da[1] < val_ext]
    elif ext == 'maximum':
    	tmp_list = [da for da in data if da[2] < val_ext]
    elif ext == 'remain':
    	tmp_list = [da for da in data if da[3] < val_ext]
            
    # sort_by의 이름에 따라 idx 값 설정. 
    if sort_by == 'remain': idx = 3 
    elif sort_by == 'maximum': idx = 2
    elif sort_by == 'date': idx = 1
    elif sort_by == 'code': idx = 0
    
    # sorted 함수의 key값으로 tmp_data[idx]를 기준으로 정렬. 
    return sorted(tmp_list, key = lambda tmp_data : tmp_data[idx])

 

# 풀이2

---

def solution(data, ext, val_ext, sort_by):      
    tmp_list = []
    if sort_by == 'remain':
        idx = 3
    elif sort_by == 'maximum':
        idx = 2
    elif sort_by == 'date':
        idx = 1
    elif sort_by == 'code':
        idx = 0

    for i in range(len(data)):
        da = data[i]
        code, date, maxi, rema = da[0], da[1], da[2], da[3]
        if ext == 'code':
            if code < val_ext: tmp_list.append(da)
            else: pass
        elif ext == 'date':
            if date < val_ext: tmp_list.append(da)
            else: pass
        elif ext == 'maximum':
            if maxi < val_ext: tmp_list.append(da)
            else: pass
        elif ext == 'remain':
            if rema < val_ext: tmp_list.append(da)
            else: pass

    return sorted(tmp_list, key = lambda tmp_data : tmp_data[idx])

""" 그닥 이쁘진 않은 코드.효율성-메모리 측면에선 꽤 좋았음. 예외처리가 빠르게 되고, 리스트
원소 내부를 내부적으로 다 들여다볼 필요가 없다보니 그런 듯함. """