파이썬 알고리즘 : 데이터 분석

정렬

2024년 4월 19일 알고리즘 문제풀이

문제

난이도

Lv. 1

코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
def solution(data, ext, val_ext, sort_by):
    answer = []
    for x in data:
        arr = dict()
        arr["code"] = x[0]
        arr["date"] = x[1]
        arr["maximum"] = x[2]
        arr["remain"] = x[3]
        if arr[ext] < val_ext:
            answer.append(x)
    if sort_by == 'code':
        idx = 0
    elif sort_by == 'date':
        idx = 1
    elif sort_by == 'maximum':
        idx = 2
    else:
        idx = 3
    answer.sort(key = lambda x: (x[idx]))
    return answer