파이썬에서 파일 이름에서 확장자 추출하기

파이썬에서 파일 이름에서 확장자를 추출하는 방법에 대해 자세히 알아보겠습니다. 파일 확장자는 파일의 종류를 나타내는 중요한 정보입니다. 파이썬의 os 모듈은 파일 시스템과 관련된 다양한 기능을 제공하며, 그중 os.path.splitext() 함수를 사용하여 파일 이름과 확장자를 간편하게 분리할 수 있습니다.

os.path.splitext() 함수

• 기능: 파일 경로를 입력받아 파일 이름과 확장자를 튜플 형태로 반환합니다.
• 사용법:

  import os
  
  file_path = "image.jpg"
  file_name, file_extension = os.path.splitext(file_path)
  
  print("파일 이름:", file_name)
  print("확장자:", file_extension)
  

• 출력:

  파일 이름: image
  확장자: .jpg
  

예시: 다양한 파일 경로 처리하기

import os

file_paths = ["data.csv", "/home/user/documents/report.pdf", "image.png"]

for file_path in file_paths:
  file_name, file_extension = os.path.splitext(file_path)
  print(f"파일 경로: {file_path}")
  print(f"파일 이름: {file_name}")
  print(f"확장자: {file_extension}")
  print("-" * 20)

추가 설명

• 파일 경로: 절대 경로 또는 상대 경로 모두 사용 가능합니다.
• 확장자: 점(.)을 포함하여 반환됩니다.
• 다른 모듈: pathlib 모듈도 파일 경로 처리에 유용하게 사용될 수 있습니다.

활용 예시

• 파일 종류별 처리: 확장자를 기반으로 파일을 분류하거나 특정 형식의 파일만 처리할 수 있습니다.
• 파일 이름 변경: 확장자를 변경하거나 파일 이름에 확장자를 추가할 수 있습니다.
• 파일 시스템 관리: 파일 목록을 생성하거나 파일 정보를 추출하는 등 다양한 파일 시스템 작업에 활용할 수 있습니다.

결론

os.path.splitext() 함수는 파이썬에서 파일 이름과 확장자를 분리하는 가장 간단하고 효과적인 방법입니다. 이 함수를 활용하여 파일 처리 관련 다양한 작업을 자동화하고 효율적으로 수행할 수 있습니다.

• 더 깊이 있는 내용: pathlib 모듈 활용, 정규 표현식을 이용한 복잡한 파일 이름 처리 등
• 실제 예시: 특정 디렉토리 내의 모든 파일의 확장자를 추출하여 카운트하는 프로그램 등

파이썬으로 파일 이름에서 확장자 추출하기: 실제 코드 예시

다양한 상황에 맞춰 활용할 수 있는 몇 가지 코드 예시를 준비했습니다.

기본적인 파일 확장자 추출

import os

file_path = "image.jpg"
file_name, file_extension = os.path.splitext(file_path)

print("파일 이름:", file_name)
print("확장자:", file_extension)

여러 파일 경로 처리하기

import os

file_paths = ["data.csv", "/home/user/documents/report.pdf", "image.png"]

for file_path in file_paths:
  file_name, file_extension = os.path.splitext(file_path)
  print(f"파일 경로: {file_path}")
  print(f"파일 이름: {file_name}")
  print(f"확장자: {file_extension}")
  print("-" * 20)

특정 디렉토리 내의 모든 파일 처리하기

import os

directory = "my_images"  # 처리할 디렉토리 지정

for filename in os.listdir(directory):
  if filename.endswith(('.jpg', '.png', '.gif')):  # 특정 확장자만 처리
    file_path = os.path.join(directory, filename)
    file_name, file_extension = os.path.splitext(file_path)
    print(f"파일 이름: {file_name}")
    print(f"확장자: {file_extension}")

파일 확장자를 기반으로 파일 분류하기

import os

directory = "my_files"
extensions = {}

for filename in os.listdir(directory):
  file_name, file_extension = os.path.splitext(filename)
  if file_extension not in extensions:
    extensions[file_extension] = []
  extensions[file_extension].append(filename)

for ext, files in extensions.items():
  print(f"확장자 {ext} 파일:")
  for file in files:
    print(file)

파일 이름 변경하기

import os

def change_extension(file_path, new_ext):
  file_name, _ = os.path.splitext(file_path)
  new_file_path = file_name + new_ext
  os.rename(file_path, new_file_path)

# 예시: image.jpg 파일을 image.png로 변경
change_extension("image.jpg", ".png")

각 코드에 대한 간략한 설명:

• 1번: 가장 기본적인 예시로, 단일 파일 경로에서 파일 이름과 확장자를 분리합니다.
• 2번: 여러 파일 경로를 리스트에 담아 반복 처리하며, 각 파일의 정보를 출력합니다.
• 3번: 특정 디렉토리 내의 파일들을 순회하며 지정한 확장자의 파일들만 처리합니다.
• 4번: 파일들을 확장자별로 분류하여 딕셔너리 형태로 저장합니다.
• 5번: 파일의 확장자를 변경하는 함수를 정의하고 사용하는 예시입니다.

위 코드들을 바탕으로 다양한 파일 처리 작업을 수행할 수 있습니다.

• 특정 확장자의 파일만 선택하여 작업: 이미지 파일만 처리하거나, 특정 문서 형식의 파일만 추출하는 등
• 파일 이름 변경: 일괄적으로 파일 이름을 변경하거나, 날짜 정보를 추가하는 등
• 파일 분류 및 정리: 파일을 종류별로 폴더에 이동하거나, 중복 파일을 찾는 등

파이썬에서 파일 이름에서 확장자 추출하기: 다른 방법들

os.path.splitext() 외에도 파이썬에서 파일 이름에서 확장자를 추출하는 다양한 방법들이 있습니다. 각 방법마다 장단점이 있으므로, 상황에 맞게 적절한 방법을 선택하는 것이 중요합니다.

문자열 슬라이싱 (String Slicing)

가장 직관적인 방법으로, 문자열의 마지막 점(.)의 위치를 찾아 그 이후의 부분을 추출합니다.

filename = "image.jpg"
extension = filename[filename.rfind('.'):]
print(extension)  # 출력: .jpg

장점: 간단하고 직관적입니다.단점: 점(.)이 여러 개 포함된 파일 이름의 경우 예상치 못한 결과를 초래할 수 있습니다.

정규 표현식 (Regular Expression)

정규 표현식을 사용하여 더욱 복잡한 패턴의 파일 이름을 처리할 수 있습니다.

import re

filename = "my_file.tar.gz"
match = re.search(r'\.(\w+)$', filename)
if match:
    extension = match.group(1)
    print(extension)  # 출력: gz

장점: 유연하고 다양한 패턴을 처리할 수 있습니다.단점: 정규 표현식에 대한 이해가 필요하며, 오버헤드가 발생할 수 있습니다.

pathlib 모듈

pathlib 모듈은 파일 시스템 경로를 객체로 다루는 더욱 객체 지향적인 방법을 제공합니다.

from pathlib import Path

filename = "image.jpg"
path = Path(filename)
extension = path.suffix
print(extension)  # 출력: .jpg

장점: os.path 모듈보다 더욱 객체 지향적이고 사용하기 편리합니다.단점: Python 3.4 이상에서만 사용 가능합니다.

각 방법의 비교

어떤 방법을 선택해야 할까요?

• 간단한 경우: os.path.splitext()를 사용하는 것이 가장 일반적이고 효율적입니다.
• 복잡한 패턴 처리: 정규 표현식을 사용하면 다양한 경우에 대응할 수 있습니다.
• 객체 지향적인 접근: pathlib 모듈을 사용하면 더욱 객체 지향적인 코드를 작성할 수 있습니다.

결론적으로, 상황에 맞는 적절한 방법을 선택하여 사용하면 됩니다.

• 특정 상황: 어떤 종류의 파일 이름을 처리하고 싶으신가요?
• 목표: 추출한 확장자를 어떻게 활용하고 싶으신가요?
• 기타 요구사항: 성능, 가독성 등

예시:

• 특정 디렉토리 내의 모든 파일의 확장자를 카운트: collections.Counter와 함께 사용하여 각 확장자별 파일 개수를 세어볼 수 있습니다.
• 특정 확장자의 파일만 선택하여 처리: if 문으로 확장자를 비교하여 원하는 파일만 처리할 수 있습니다.