[파이썬] 파일 경로의 확장자로 필터링

파일 경로에서 특정 확장자를 가진 파일을 필터링하는 것은 Python에서 매우 일반적인 작업입니다. 이번 블로그 포스트에서는 이 과정을 간단한 코드로 설명하고 예시를 보여드리겠습니다.

파일 경로의 확장자로 필터링하기

파이썬의 os 모듈을 사용하면 파일 및 디렉토리 조작에 대한 다양한 기능을 활용할 수 있습니다. 우리의 목적에 맞게 파일 경로의 확장자로 필터링해 보겠습니다.

다음은 파일 경로에서 확장자를 필터링하는 함수의 예시입니다.

import os

def filter_files_by_extension(path, extension):
    filtered_files = []
    for file_name in os.listdir(path):
        if file_name.endswith(extension):
            filtered_files.append(file_name)
    return filtered_files

위의 코드에서 filter_files_by_extension 함수는 두 개의 매개변수를 받습니다. 첫 번째 매개변수 path는 파일 경로를 나타내며, 두 번째 매개변수 extension은 필터링할 확장자입니다.

함수는 주어진 경로의 모든 파일을 반복하며, 파일 이름이 주어진 확장자로 끝나는 경우에만 filtered_files 리스트에 추가합니다. 이후 filtered_files 리스트를 반환합니다.

사용 예시

위에서 정의한 filter_files_by_extension 함수를 사용하여 특정 경로에서 특정 확장자를 가진 파일을 필터링하는 방법을 살펴보겠습니다.

folder_path = '/path/to/folder'
file_extension = '.txt'

filtered_files = filter_files_by_extension(folder_path, file_extension)

for file in filtered_files:
    print(file)

위의 예시에서는 folder_path 변수에 필터링할 폴더 경로를, file_extension 변수에 필터링할 확장자를 입력합니다. 그런 다음 filter_files_by_extension 함수를 호출하여 filtered_files 리스트를 가져옵니다.

이후 filtered_files 리스트를 반복하면서 각 파일 이름을 출력합니다.

요약

Python을 사용하여 파일 경로의 확장자로 필터링하는 방법에 대해 알아보았습니다. os 모듈을 사용하여 파일 경로의 확장자를 확인하고 필요한 작업을 수행할 수 있습니다. 실제로 확장자로 필터링할 때는 용도에 맞게 코드를 수정해야 할 수도 있습니다. 이 예제는 단지 기본적인 개념을 보여주기 위한 것입니다.

이제 파일 경로에서 확장자로 필터링하는 방법에 대해 더욱 익숙해졌을 것입니다. 이 기술은 특정 유형의 파일을 처리하고 데이터를 추출하거나 분석하는 데 유용합니다. 적용하고자 하는 작업에 따라 코드를 조정하여 유연하게 사용할 수 있습니다.