[Python/파이썬] os.walk()를 사용한 디렉토리 및 파일 탐색하기

Python의 os.walk() 함수는 파일 시스템을 탐색하여 디렉토리와 파일을 순회하는 데 사용됩니다. 이 함수는 재귀적으로 디렉토리를 탐색하며, 디렉토리 내의 모든 하위 디렉토리와 파일을 방문합니다. 아래에서는 os.walk() 함수를 사용하는 여러 예제를 살펴보겠습니다.

예제 1: 현재 디렉토리의 모든 파일과 디렉토리 출력

import os

for root, dirs, files in os.walk('.'):
    print(f'현재 디렉토리: {root}')
    print(f'하위 디렉토리: {dirs}')
    print(f'파일: {files}')
    print()

이 예제는 현재 디렉토리부터 시작하여 모든 하위 디렉토리와 파일을 출력합니다.

예제 2: 특정 디렉토리에서 특정 확장자를 가진 파일만 출력

import os

for root, dirs, files in os.walk('.', topdown=True):
    for name in files:
        if name.endswith('.txt'):
            print(os.path.join(root, name))

특정 디렉토리에서 특정 확장자를 가진 파일들만 출력하는 예제입니다.

예제 3: 하위 디렉토리까지 모두 탐색하여 파일 출력

import os

for root, dirs, files in os.walk('.'):
    for file in files:
        print(os.path.join(root, file))

하위 디렉토리까지 모두 탐색하여 모든 파일을 출력하는 예제입니다.

예제 4: 특정 디렉토리에서 특정 파일명을 가진 파일의 경로 출력

import os

for root, dirs, files in os.walk('.'):
    for file in files:
        if file == 'example.txt':
            print(os.path.join(root, file))

특정 디렉토리에서 특정 파일명을 가진 파일의 경로를 출력하는 예제입니다.

예제 5: 하위 디렉토리를 제외하고 파일만 출력

import os

for root, dirs, files in os.walk('.'):
    if root != '.':
        continue
    for file in files:
        print(os.path.join(root, file))

하위 디렉토리를 제외하고 현재 디렉토리의 파일만 출력하는 예제입니다.

예제 6: 디렉토리를 제외하고 파일만 출력

import os

for root, dirs, files in os.walk('.'):
    if root != '.':
        break
    for file in files:
        print(os.path.join(root, file))

디렉토리를 제외하고 현재 디렉토리의 파일만 출력하는 예제입니다.