¿Cómo obtener la extensión de archivo en Python?

En este artículo, cubriremos Cómo extraer extensiones de archivo usando Python, Obtener extensión de archivo en Python, podemos usar cualquiera de los dos enfoques diferentes que se analizan a continuación:

• Use el módulo os.path para extraer la extensión del archivo en Python
• Use el módulo pathlib para extraer la extensión del archivo en Python

Método 1: Usar la función splitext() del módulo os de Python

Esta función divide la string de la ruta del archivo en el nombre del archivo y la extensión del archivo en un par de raíz y extensión, de modo que cuando ambos se agregan, podemos recuperar la ruta del archivo nuevamente (nombre_archivo + extensión = ruta). Se prefiere el uso de esta función cuando el módulo del sistema operativo ya se está utilizando.

import os
 
# this will return a tuple of root and extension
split_tup = os.path.splitext('my_file.txt')
print(split_tup)
 
# extract the file name and extension
file_name = split_tup[0]
file_extension = split_tup[1]
 
print("File Name: ", file_name)
print("File Extension: ", file_extension)

Producción:

('my_file', '.txt')
File Name:  my_file
File Extension:  .txt

Método 2: Usar el módulo Pathlib

El método pathlib.Path().suffix del módulo Pathlib se puede utilizar para extraer la extensión de la ruta del archivo. Este método es el preferido para un enfoque orientado a objetos.

import pathlib
 
# function to return the file extension
file_extension = pathlib.Path('my_file.txt').suffix
print("File Extension: ", file_extension)

Producción:

File Extension:  .txt