프로그래머스 [lv2] 파일명 정렬 파이썬

정답 코드

import re


def solution(files):
    filtered_files = [re.split("([0-9]+)", file) for file in files]
    sorted_files = sorted(filtered_files, key=lambda x: (x[0].lower(), int(x[1])))
    return [(''.join(s)) for s in sorted_files]
    

files = ["img12.png", "img10.png", "img02.png", "img1.png", "IMG01.GIF", "img2.JPG"]
solution(files)
>>> ['img1.png', 'IMG01.GIF', 'img02.png', 'img2.JPG', 'img10.png', 'img12.png']

 

📕 정규 표현식으로 필터링 후 : filtered_files

def solution(files):

    filtered_files = [re.split("([0-9]+)", file) for file in files]
    print(fileted_files)
    >>> [['img', '12', '.png'], ['img', '10', '.png'], ['img', '02', '.png'], ['img', '1', '.png'], ['IMG', '01', '.GIF'], ['img', '2', '.JPG']]

우선 head 와 number 를 나눠야 합니다. head 는 문자만, number 는 숫자만 들어갑니다.

➡️ tail 부분에 number 가 들어 갈 수 있지만, Head, Number 만 나누면 되기 때문에 이 부분에만 집중하면 됩니다.

정규 표현식을 이용하여 숫자가 들어간 부분을 기준으로 분리합니다. 

 

📗 람다식으로 정렬

def solution(files):

    filtered_files = [re.split("([0-9]+)", file) for file in files]

    sorted_files = sorted(filtered_files, key=lambda x: (x[0].lower(), int(x[1])))
    print(sorted_files)
    >>> [['img', '1', '.png'], ['IMG', '01', '.GIF'], ['img', '02', '.png'], ['img', '2', '.JPG'], ['img', '10', '.pn', '21', 'g'], ['img', '12', '.png']]

1️⃣ 첫번째 Head 정렬

람다식과 sorted() 메소드를 이용하여 정렬 조건을 구현할 수  있습니다.

우선 filtered_files 의 모든 원소들의 첫번째 인자끼리 비교 후 오름차순 정렬합니다. (sorted()의 세번째 인자로 reverese=True /False 조건을 통해 오름차순/내림차순을 설정할 수 있습니다)

 

2️⃣ 두번째 Number 정렬

첫번째 인자끼리의 비교로 head 끼리 비교를 마쳤다면, lambda 의 두번째 인자로 number 를 비교할 수 있습니다.

🦁 만약 문제 조건이 number 는 내림차순 정렬이었다면, int(x[1]) 앞에 - (minus) 부호를 붙여 내림차순을 구현할 수 있습니다. 해당 문제는 오름차순 정렬이기 때문에 부호를 써주지 않아도 됩니다.