[python] [3차] 파일명 정렬 lv.2

프로그래머스

아이디어

files 배열로 들어오는 파일명을 커스텀한 기준으로 정렬해주면 됩니다.

filename_split 함수에서 원하는 방식으로 문자열을 head와 number로 쪼개면서 형변환을 해주었고 그걸 기준으로 오름차순 정렬해주었습니다.

파이썬은 sorted가 범용적으로 사용하기에 매우 편하게 되어있습니다. 사용성 면에서 정말 설계가 잘 된 언어라는게 종종 느껴집니다.

[sort] 파이썬에서 커스텀한 기준으로 이터러블한 객체 정렬하기

 

코드

def solution(files):

    def filename_split(fn) :
        head = ''
        number = ''
        
        flag = 0
        for c in fn :
            if ord(c) >= ord('0') and ord(c) <= ord('9') :
                flag = 1
                number+=c
            elif flag == 0 :
                head+=c
            else : break
        return head.lower(), int(number)
    
    return sorted(files, key = lambda x : filename_split(x))