문제: 프로그래머스 PCCE기출문제 10번 데이터 분석

def solution(data, ext, val_ext, sort_by):
    answer = [[]]
    a = []
    sort = ["code", "date", "maximum", "remain"]
    for i in range(len(data)):
        for index, c in enumerate(sort):
            if ext == c:
                if data[i][index] < val_ext:
                    a.append(data[i])
    for num, s in enumerate(sort):
        if s == sort_by:
            answer = sorted(a, key=lambda x:x[num])


    return answer

sorted(a, key=lambda x:x[num]) 

-> 배열a에서 원소x의 x[num]번째 원소를 기준으로 오름차순 정렬한 결과를 반환한다. 

반응형

+ Recent posts