Extraire l'extension du nom de fichier en Python

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

Oui. Utilisation os.path.splitext(voir la documentation Python 2.X ou la 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 fractionnement de chaînes manuelles, os.path.splitexttraitera correctement /a/b.c/dcomme n'ayant pas d'extension au lieu d'avoir une extension .c/d, et il traitera .bashrccomme 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', '')

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

Nouveau dans la version 3.4.

import pathlib

print(pathlib.Path('yourPath.example').suffix) # '.example'

Je suis surpris que personne ne l'ait pathlibencore mentionné , c'est pathlibgénial!

Si vous avez besoin de tous les suffixes (par exemple, si vous en avez un .tar.gz), vous .suffixesen retournerez une liste!

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

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

Une option peut être la séparation du point:

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

Aucune 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

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

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

N'importe laquelle des solutions ci-dessus fonctionne, mais sur Linux, j'ai trouvé qu'il y a une nouvelle ligne à la fin de la chaîne d'extension qui empêchera les correspondances de réussir. Ajoutez la strip()méthode à la fin. Par exemple:

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

Avec splitext il y a des problèmes avec les fichiers à double extension (par exemple file.tar.gz, file.tar.bz2, etc ..)

>>> fileName, fileExtension = os.path.splitext('/path/to/somefile.tar.gz')
>>> fileExtension 
'.gz'

mais devrait être: .tar.gz

Les solutions possibles sont ici

Vous pouvez trouver de bonnes choses dans le module pathlib (disponible en python 3.x).

import pathlib
x = pathlib.PurePosixPath("C:\\Path\\To\\File\\myfile.txt").suffix
print(x)

# Output 
'.txt'

Bien qu'il s'agisse d'un sujet ancien, mais je me demande pourquoi il n'y a aucun mention d'une API très simple de python appelée rpartition dans ce cas:

pour obtenir l'extension d'un chemin absolu de fichier donné, vous pouvez simplement taper:

filepath.rpartition('.')[-1]

exemple:

path = '/home/jersey/remote/data/test.csv'
print path.rpartition('.')[-1]

vous donnera: 'csv'

Juste jointous pathlib suffixes.

>>> x = 'file/path/archive.tar.gz'
>>> y = 'file/path/text.txt'
>>> ''.join(pathlib.Path(x).suffixes)
'.tar.gz'
>>> ''.join(pathlib.Path(y).suffixes)
'.txt'

Surpris, cela n'a pas encore été mentionné:

import os
fn = '/some/path/a.tar.gz'

basename = os.path.basename(fn)  # os independent
Out[] a.tar.gz

base = basename.split('.')[0]
Out[] a

ext = '.'.join(basename.split('.')[1:])   # <-- main part

# if you want a leading '.', and if no result `None`:
ext = '.' + ext if ext else None
Out[] .tar.gz

Avantages:

• Fonctionne comme prévu pour tout ce à quoi je peux penser
• Aucun module
• Aucun regex
• Multiplateforme
• Facilement extensible (par exemple, pas de points de tête pour l'extension, uniquement la dernière partie de l'extension)

Comme fonction:

def get_extension(filename):
    basename = os.path.basename(filename)  # os independent
    ext = '.'.join(basename.split('.')[1:])
    return '.' + ext if ext else None

Vous pouvez utiliser un splitsur filename:

f_extns = filename.split(".")
print ("The extension of the file is : " + repr(f_extns[-1]))

Cela ne nécessite pas de bibliothèque supplémentaire

filename='ext.tar.gz'
extension = filename[filename.rfind('.'):]

Il s'agit d'une technique de représentation directe des chaînes: je vois beaucoup de solutions mentionnées, mais je pense que la plupart envisagent le split. Split le fait cependant à chaque occurrence de "." . Ce que vous préférez, c'est la partition.

string = "folder/to_path/filename.ext"
extension = string.rpartition(".")[-1]

Une autre solution avec split droit:

# to get extension only

s = 'test.ext'

if '.' in s: ext = s.rsplit('.', 1)[1]

# or, to get file name and extension

def split_filepath(s):
    """
    get filename and extension from filepath 
    filepath -> (filename, extension)
    """
    if not '.' in s: return (s, '')
    r = s.rsplit('.', 1)
    return (r[0], r[1])

Même cette question est déjà répondue, j'ajouterais la solution dans Regex.

>>> import re
>>> file_suffix = ".*(\..*)"
>>> result = re.search(file_suffix, "somefile.ext")
>>> result.group(1)
'.ext'

Un vrai one-liner, si vous aimez les regex. Et cela n'a pas d'importance même si vous avez des "." Supplémentaires. au milieu

import re

file_ext = re.search(r"\.([^.]+)$", filename).group(1)

Voir ici pour le résultat: Cliquez ici

Il s'agit de la méthode la plus simple pour obtenir à la fois le nom de fichier et l'extension en une seule ligne .

fName, ext = 'C:/folder name/Flower.jpeg'.split('/')[-1].split('.')

>>> print(fName)
Flower
>>> print(ext)
jpeg

Contrairement à d'autres solutions, vous n'avez pas besoin d'importer de package pour cela.

Pour les drôles ... il suffit de rassembler les extensions dans un dict et de les suivre toutes dans un dossier. Tirez ensuite les extensions que vous souhaitez.

import os

search = {}

for f in os.listdir(os.getcwd()):
    fn, fe = os.path.splitext(f)
    try:
        search[fe].append(f)
    except:
        search[fe]=[f,]

extensions = ('.png','.jpg')
for ex in extensions:
    found = search.get(ex,'')
    if found:
        print(found)

essaye ça:

files = ['file.jpeg','file.tar.gz','file.png','file.foo.bar','file.etc']
pen_ext = ['foo', 'tar', 'bar', 'etc']

for file in files: #1
    if (file.split(".")[-2] in pen_ext): #2
        ext =  file.split(".")[-2]+"."+file.split(".")[-1]#3
    else:
        ext = file.split(".")[-1] #4
    print (ext) #5

1. obtenir tous les noms de fichiers dans la liste
2. diviser le nom de fichier et vérifier l'avant-dernière extension, est-ce dans la liste pen_ext ou non?
3. si oui, joignez-le à la dernière extension et définissez-le comme extension du fichier
4. sinon, il suffit de mettre la dernière extension comme extension du fichier
5. puis vérifiez-le

# try this, it works for anything, any length of extension
# e.g www.google.com/downloads/file1.gz.rs -> .gz.rs

import os.path

class LinkChecker:

    @staticmethod
    def get_link_extension(link: str)->str:
        if link is None or link == "":
            return ""
        else:
            paths = os.path.splitext(link)
            ext = paths[1]
            new_link = paths[0]
            if ext != "":
                return LinkChecker.get_link_extension(new_link) + ext
            else:
                return ""

def NewFileName(fichier):
    cpt = 0
    fic , *ext =  fichier.split('.')
    ext = '.'.join(ext)
    while os.path.isfile(fichier):
        cpt += 1
        fichier = '{0}-({1}).{2}'.format(fic, cpt, ext)
    return fichier

name_only=file_name[:filename.index(".")

Cela vous donnera le nom du fichier jusqu'au premier ".", Qui serait le plus courant.