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

문제 링크

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

문제 설명

data에서 ext 값이 val_ext보다 작은 데이터만 뽑은 후, sort_by에 해당하는 값을 기준으로 오름차순으로 정렬하여 return 하도록 solution 함수를 완성해 주세요.

이 문제 마치 sql...

1차 시도

def solution(data, ext, val_ext, sort_by):
    column = ['code','date','maximum','remain']
    idx = column.index(ext)
    tmp = []
    for n,i in enumerate(data):
        if i[idx]<val_ext:
            tmp.append(i)
    tmp.sort(key=lambda x: x[column.index(sort_by)])
    return tmp

2차원 리스트에서 특정 데이터로 정렬하는 게 중요한 문제였다. key lambda를 쓸 줄 알아야하는!!!
근데 항상 lambda를 검색해서 써서 제대로 모르고있다.
그러니까 한 번 정리하고 넘어가야겠다.

정리

• lambda: 정렬할 때 key=lambda~를 이용해 특정 조건 기준으로 정렬이 가능하다.