[프로그래머스] PCCP 기출문제 10번

문제

2차원 배열 data에서 ext열에 해당하는 값이 val_ext보다 작은 행들을 sort_by에 해당하는 열 기준 오름차순으로 정렬시켜 가공한 새로운 data를 반환해라.
참고로 각 열의 카테고리는 다음과 같다.
['code', 'date', 'maximum', 'remain']

ex)

data = [
	[1, 20300104, 100, 80], 
	[2, 20300804, 847, 37], 
	[3, 20300401, 10, 8]
]
ext = 'date'
val_ext = 20300501
sort_by = 'remain'

전체코드

def solution(data, ext, val_ext, sort_by):
    answer = []
    indexHeader = ['code', 'date', 'maximum', 'remain']

    data = sorted(data, key = lambda x : x[indexHeader.index(sort_by)])

    for row in data:
        if row[indexHeader.index(ext)] >= val_ext:
            continue

        answer.append(row)

    return answer