Extraction de l'extension d'un nom de fichier en Python

Question

Existe-t-il une fonction permettant d'extraire l'extension d'un nom de fichier ?

Answer 1

Oui. Utilisez os.path.splitext (voir Documentation Python 2.X o Documentation Python 3.X ):

>>> import os
>>> filename, file_extension = os.path.splitext('/path/to/somefile.ext')
>>> filename
'/path/to/somefile'
>>> file_extension
'.ext'

Contrairement à la plupart des tentatives de séparation manuelle des chaînes de caractères, os.path.splitext traitera correctement /a/b.c/d comme n'ayant pas d'extension au lieu d'avoir une extension .c/d et il traitera .bashrc comme n'ayant pas d'extension au lieu d'avoir une extension .bashrc :

>>> os.path.splitext('/a/b.c/d')
('/a/b.c/d', '')
>>> os.path.splitext('.bashrc')
('.bashrc', '')

Answer 2

import os.path
extension = os.path.splitext(filename)[1]

Answer 3

import os.path
extension = os.path.splitext(filename)[1][1:]

Pour obtenir uniquement le texte de l'extension, sans le point.

Answer 4

Pour les cas d'utilisation simples, une option peut être de se séparer de dot :

>>> filename = "example.jpeg"
>>> filename.split(".")[-1]
'jpeg'

Pas d'erreur lorsque le fichier n'a pas d'extension :

>>> "filename".split(".")[-1]
'filename'

Mais vous devez faire attention :

>>> "png".split(".")[-1]
'png'    # But file doesn't have an extension

Il ne fonctionne pas non plus avec les fichiers cachés des systèmes Unix :

>>> ".bashrc".split(".")[-1]
'bashrc'    # But this is not an extension

Pour un usage général, préférez os.path.splitext

Answer 5

Cela vaut la peine d'ajouter un bas de page pour que vous ne vous demandiez pas pourquoi les JPG n'apparaissent pas dans votre liste.

os.path.splitext(filename)[1][1:].strip().lower()