파이썬에서 현재 디렉토리, 하위 디렉토리 이름, 개수 가져오기

 

os.walk 모듈을 이용하여 현재 디렉토리명, 하위 디렉토리명, 현재 디렉토리에 있는 파일명을 가져오는 방법입니다.

하위 디렉토리명, 파일명과 더불어 개수도 가져올 수 있습니다.

 

os.walk로 현재 디렉토리 정보 가져오기

import os

print (next(os.walk('/home/jrkim/mlflow/')))

 

os.walk 호출 시 정보를 가져오고 싶은 디렉토리를 입력하면 다음과 같은 리스트로 리턴해줍니다.

('/home/jrkim/mlflow/', ['mllogs', '300seconds', 'mlflow_git', 'mlruns'], ['conda.yaml', 'subdir.py', 'b', 'a.py'])

리스트의 첫번째 항목이 현재 디렉토리, 두번째 항목이 하위 디렉토리명 리스트, 세번째 항목이 파일명 리스트 입니다.

해당 인덱스로 접근하면 값을 가져 올 수 있습니다.

 

하위 디렉토리명, 파일명과 개수 가져오기

import os

print (len(next(os.walk('/home/jrkim/mlflow/'))[1]), next(os.walk('/home/jrkim/mlflow/'))[1])
print (len(next(os.walk('/home/jrkim/mlflow/'))[2]), next(os.walk('/home/jrkim/mlflow/'))[2])

 

위 코드를 실행하면 아래와 같이 출력됩니다.

4 ['mllogs', '300seconds', 'mlflow_git', 'mlruns']
4 ['conda.yaml', 'subdir.py', 'b', 'a.py']

두번째 인덱스로 접근하면 하위 디렉토리명을 가져올 수 있습니다. len() 함수를 사용하면 리스트에 들어 있는 값의 길이를 구해주므로 하위 디렉토리의 개수를 나타냅니다.

세번째 인덱스로 접근하면 현재 디렉토리에 존재하는 파일명을 가져 올 수 있습니다. 마찬가지로 len 함수로 파일 개수를 알 수 있습니다.

 

이상 파이썬에서 하위 디렉토리명과 개수를 가져오는 방법을 알아보았습니다.