Python - 파일 개수 확인

특정 디렉토리의 파일 개수를 확인하는 방법을 소개합니다.

1. os.listdir()를 이용한 방법

os.listdir(dir)은 dir 하위의 모든 파일 및 디렉토리를 리스트로 가져옵니다. 그리고 len()으로 리스트의 크기를 계산하면, 파일들의 개수를 가져올 수 있습니다.

이 방법은, 특정 디렉토리의 바로 아래에 있는 파일/디렉토리 개수만 계산할 수 있으며, 하위 디렉토리 아래의 파일 개수까지 계산하진 않습니다.

import os

dir_path = "./"

file_list = os.listdir(dir_path)
size = len(file_list)
print(size)

Output:

88

2. os.walk()를 이용한 방법

os.walk(dir)는 dir 경로 아래의 모든 디렉토리와 파일을 탐색합니다.

하위의 모든 파일 및 디렉토리 개수를 계산하고 싶다면 이 방법을 사용하시면 됩니다.

아래와 같이 탐색된 디렉토리와 파일들의 개수를 합하여, 전체 파일 개수와 디렉토리 개수를 계산할 수 있습니다. (파일 이름 및 디렉토리 이름을 확인하고 싶으면 주석의 코드를 사용하시면 됩니다.)

import os

dir_cnt = 0
file_cnt = 0
dir_path = "./"

for (root, dirs, files) in os.walk(dir_path):
    if len(dirs) > 0:
        dir_cnt = dir_cnt + len(dirs)
        # for d in dirs:
        #     print(d)
    if len(files) > 0:
        file_cnt = file_cnt + len(files)
        # for f in files:
        #     print(f)

print(f"dir cnt: {dir_cnt}")
print(f"file cnt: {file_cnt}")

Output:

dir cnt: 2
file cnt: 92

os.walk()에 대한 자세한 사용 방법은 Python - os.walk()를 사용하여 디렉토리, 파일 탐색를 참고해주세요.

Loading script...
codechachaCopyright ©2019 codechacha