[Python] os 모듈의 유용한 함수 정리

os 모듈은 OS(Operating System)를 제어할 수 있는 유용한 모듈이다.

프로젝트를 진행하면서 많이 사용한 함수를 정리해보려고 한다.

 

import os

먼저 os 모듈은 임포트해준다.

 

1.

os.getcwd()

get curren working directory의 약자로, 현재 작업 디렉토리를 반환한다.

상대 경로 (ex. ../data/train/)를 지정하거나 현재 디렉토리를 확인할 때 유용하다.

리눅스 터미널 창에서 pwd 명령어와 같은 역할을 한다.

 

 

2.

os.listdir('./')

인자로 './' 를 넣어주면 현재 디렉토리에 있는 모든 파일과 폴더를 보여준다.

이 때 리눅스 터미널 창에서 ls 명령어와 같은 역할을 한다.

결과는 리스트 형태로 주어지기 때문에, 하위 디렉토리의 더 하위 디렉토리 등을 확인하거나 거기서 작업해야 할 게 있을 때, 반복문으로 사용하면 편리하다.

 

 

3.

os.path.join('Desktop', 'testdir')

각 인자의 경로(path)를 하나로 합쳐(join)준다.

join된 path는 문자열 형태이므로 해당 경로는 실제 존재할 수도 있고, 안 할 수도 있다. 따라서 확인이 필요하다.

1,2,3을 모두 응용하면 아래와 같다.

 

for path in os.listdir('./'):
	path_name = os.path.join(os.getcwd(), path)
	print(path_name)

 

4.

img_path = '../image1.jpg'
os.path.exists(img_path)

절대/상대 경로, 혹은 위와 같은 방법으로 만든 경로가 실제 존재하는지 확인하는 함수다.

 

 

5.

img_path = '../image1.jpg'
os.path.isfile(img_path)

dir_path = 'testdir/'
os.path.isdir(dir_path)

해당 경로가 디렉토리인지 파일인지 확인하는 함수다.

isfile은 파일인지, isdir은 디렉토리인지를 알려주며, 존재하지 않는 경로는 False로 출력된다.

 

 

6.

os.mkdir(dir_name)

make directory, 디렉토리를 생성하는 함수다.

위의 예시에서 test_mkdir 이라는 디렉토리를 생성해봤다.

 

 

---

os 모듈 중에서도 기본 중의 기본인 함수, 또 이번 프로젝트를 진행하며 유용하게 사용했던 함수를 정리했다.

나중에 'os에 이런 기능이 있었는데..' 하고 다시 찾아보지 않도록 직접 써본다.