본문으로 건너뛰기

Python에서 확장자별로 파일을 필터링하는 방법

파이썬에서 확장자로 파일을 필터링하는 방법.

Python에서 파일을 확장자로 필터링하는 방법에 대한 단계별 튜토리얼입니다:

단계 1 : os 모듈 가져오기

파일 및 디렉토리를 다루기 위해 내장 os 모듈을 가져와야합니다. 이 모듈은 운영 체제와 상호 작용하기 위한 다양한 함수를 제공합니다.

import os

단계 2 : 디렉토리 경로 정의

파일을 필터링하려는 디렉토리 경로를 지정합니다. 절대 경로 또는 상대 경로를 제공할 수 있습니다.

directory_path = '/경로/디렉토리'

단계 3 : 디렉토리의 모든 파일 나열

os.listdir() 함수를 사용하여 지정된 디렉토리에있는 모든 파일과 디렉토리의 목록을 가져옵니다.

files = os.listdir(directory_path)

단계 4 : 확장자로 파일 필터링

파일 목록을 반복하고 os.path.splitext() 함수를 사용하여 파일 이름과 확장자를 분리합니다. 확장자를 원하는 확장자와 비교하기 위해 if 문을 사용합니다.

desired_extension = '.txt'  # 원하는 확장자로 변경

filtered_files = []
for file in files:
_, extension = os.path.splitext(file)
if extension == desired_extension:
filtered_files.append(file)

또는 더 간결한 방법으로 같은 결과를 얻기 위해 리스트 컴프리헨션을 사용할 수 있습니다:

filtered_files = [file for file in files if os.path.splitext(file)[1] == desired_extension]

단계 5 : 필터링 된 파일 출력 또는 처리

필터링 된 파일 이름을 출력하거나 요구 사항에 따라 추가 처리를 수행 할 수 있습니다.

for file in filtered_files:
print(file)

이것으로 확장자로 필터링 된 파일 목록을 얻었습니다.

여기 완성 된 코드입니다:

import os

directory_path = '/경로/디렉토리'
files = os.listdir(directory_path)

desired_extension = '.txt' # 원하는 확장자로 변경

filtered_files = [file for file in files if os.path.splitext(file)[1] == desired_extension]

for file in filtered_files:
print(file)

desired_extension 변수를 필터링하려는 확장자와 일치하도록 자유롭게 수정하십시오.