[python] 특정 확장자를 가진 파일 필터링하기

파일을 다루거나 조작할 때, 특정 확장자를 가진 파일만 선택해서 사용해야 하는 경우가 있습니다. 이때 Python을 사용하여 해당 작업을 쉽게 수행할 수 있습니다.

glob 모듈을 사용하여 파일 필터링하기

Python의 glob 모듈을 사용하면 파일들을 쉽게 필터링할 수 있습니다. 이 모듈은 와일드카드를 사용하여 파일을 찾는 데에 유용합니다.

아래는 glob 모듈을 사용하여 Python에서 .txt 확장자를 가진 파일들을 필터링하는 간단한 예제입니다.

import glob

# 현재 디렉토리에서 .txt 파일들을 모두 찾기
txt_files = glob.glob('*.txt')

print(txt_files)

위 코드를 실행하면 현재 디렉토리에서 .txt 확장자를 가진 모든 파일들의 리스트를 얻을 수 있습니다.

glob 모듈을 사용하면 특정 디렉토리나 파일 패턴에 맞는 파일들을 쉽게 필터링할 수 있습니다.

os 모듈을 사용하여 파일 필터링하기

os 모듈을 사용하여 특정 디렉토리 안에 있는 모든 파일을 가져 올 수 있습니다. 이후 리스트 컴프리헨션을 사용하여 원하는 확장자를 가진 파일들만 골라낼 수 있습니다.

import os

# 현재 디렉토리에서 모든 파일들을 가져오기
all_files = os.listdir()

# .txt 확장자를 가진 파일들만 골라내기
txt_files = [file for file in all_files if file.endswith('.txt')]

print(txt_files)

위와 같은 방법으로 os 모듈을 사용하여도 특정 확장자를 가진 파일들을 필터링할 수 있습니다.

결론

Python을 사용하여 파일을 필터링하는 방법은 여러 가지가 있지만, glob 모듈과 os 모듈은 가장 일반적으로 사용되는 방법입니다. 필요에 따라 적합한 방법을 선택하여 사용하면 됩니다.

참고 자료: