Haribo ML, AI, MATH, Algorithm

[3차] 파일명 정렬


[3차] 파일명 정렬

import re
def solution(files):
    file_dict = {}
    for file in files :
        digit = re.findall('\d+', file)[0]
        file_dict[file] = [file[:file.find(digit)].lower(), int(digit)]
    return [x[0] for x in sorted(file_dict.items(), key = lambda x : (x[1][0], x[1][1]))]

풀이

re.findall로 숫자가 처음 나오는 인덱스값을 구한다.

{filename : [head, digit, index]} 사전을 만든다.

(head, index)key로 정렬된 filenamereturn


Similar Posts

이전 포스트 [3차] 압축

다음 포스트 소수 만들기

Comments