python natsort로 파일을 "이름 순으로" 정렬해보자

python으로 데이터를 다룰 때 파일 경로의 리스트를 읽어오는 불러오는 경우가 많습니다. 저는 파일 이름을 순서대로 정렬해 가져와야 했습니다. 이 단순한 작업이 겉으로는 쉬워 보이지만, 처음에는 많이 헤매었습니다. 그래서 해결방법을 공유하고자 합니다.

 

시간이 없으신 분들은 가장 아래 코드만 보시면 됩니다.

---

파일 경로를 리스트로 받아올 때 문제점

예시 폴더에 1부터 22까지 번호가 매겨진 jpg 파일을 넣어놓았습니다.

data
- 1.jpg
- 2.jpg
- 3.jpg
...
- 20.jpg
- 21.jpg
- 22.jpg

 

python 내에서 이 파일들의 경로를 순차적으로 가져오기위해 os.listdir를 이용해봅니다.

import os
a = os.listdir('data')
print(sorted(a))

그럼 결과값은 다음과 같습니다.

['0.jpg', '1.jpg', '10.jpg', '11.jpg', '12.jpg', '13.jpg', '14.jpg', '15.jpg', '16.jpg', '17.jpg', '18.jpg', '19.jpg', '2.jpg', '20.jpg', '21.jpg', '22.jpg', '3.jpg', '4.jpg', '5.jpg', '6.jpg', '7.jpg', '8.jpg', '9.jpg']

이상한 점을 눈치채셨나요? 내부적으로 제일 왼쪽 글자부터 정렬하다 보니 1 다음에 2가 아닌 10, 11이 먼저 왔습니다. 이번엔 경로를 다루는 다른 패키지 glob을 사용해보겠습니다.

import glob
a = glob.glob('data/*.jpg')
print(sorted(a))

['data/0.jpg', 'data/1.jpg', 'data/10.jpg', 'data/11.jpg', 'data/12.jpg', 'data/13.jpg', 'data/14.jpg', 'data/15.jpg', 'data/16.jpg', 'data/17.jpg', 'data/18.jpg', 'data/19.jpg', 'data/2.jpg', 'data/20.jpg', 'data/21.jpg', 'data/22.jpg', 'data/3.jpg', 'data/4.jpg', 'data/5.jpg', 'data/6.jpg', 'data/7.jpg', 'data/8.jpg', 'data/9.jpg']

역시나 마찬가지로 원하는 결과가 나오지 않습니다.

 

여기서 아마 sorted(a)에 key 파라미터로 정렬 기준을 설정하시는 분들이 많으실 겁니다. 예를들어 아래와 같이 말이죠.

import os
a = os.listdir('data')
print(sorted(a, key=lambda x: int(x.split('.')[0])))

['0.jpg', '1.jpg', '2.jpg', '3.jpg', '4.jpg', '5.jpg', '6.jpg', '7.jpg', '8.jpg', '9.jpg', '10.jpg', '11.jpg', '12.jpg', '13.jpg', '14.jpg', '15.jpg', '16.jpg', '17.jpg', '18.jpg', '19.jpg', '20.jpg', '21.jpg', '22.jpg']

이렇게 하니 원하는 결과가 나왔습니다! 그러나 이건 어디까지나 파일명이 오로지 숫자만으로 이루어졌을 때만 가능합니다. 파일명에 접두어나 접미어가 붙거나, 여러 디렉터리의 데이터를 모두 불러올 때에는 key에 어떻게 함수를 입력해야 할지 생각만 해도 어질어질합니다.

 

 

해결방법

python의 natsort 라이브러리를 이용해봅시다.

pip install natsort
#pip3 install natsort

import os
import natsort

a = os.listdir('data')
print(natsort.natsorted(os.listdir('data')))

['0.jpg', '1.jpg', '2.jpg', '3.jpg', '4.jpg', '5.jpg', '6.jpg', '7.jpg', '8.jpg', '9.jpg', '10.jpg', '11.jpg', '12.jpg', '13.jpg', '14.jpg', '15.jpg', '16.jpg', '17.jpg', '18.jpg', '19.jpg', '20.jpg', '21.jpg', '22.jpg']

이렇게 하면 원하는 결과가 나옵니다. 파일 이름에 접두어나 접미어가 붙어도 상관없습니다.

['a-0.jpg', 'a-1.jpg', 'a-21.jpg', 'a-22.jpg', 'b-17.jpg', 'b-18.jpg', 'b-19.jpg', 'b-20.jpg']