Estrazione dell'estensione dal nome file in Python

Esiste una funzione per estrarre l'estensione da un nome file?

Sì. Utilizzare os.path.splitext(consultare la documentazione di Python 2.X o la documentazione di Python 3.X ):

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

A differenza della maggior parte dei tentativi di suddivisione manuale delle stringhe, os.path.splitexttratterà correttamente /a/b.c/dcome non avere estensione invece di avere estensione .c/de tratterà .bashrccome non avere estensione invece di avere estensione .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]

Novità nella versione 3.4.

import pathlib

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

Sono sorpreso che nessuno abbia pathlibancora menzionato , pathlibÈ fantastico!

Se hai bisogno di tutti i suffissi (ad esempio se hai un .tar.gz), .suffixesverrà restituito un elenco di essi!

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

Per ottenere solo il testo dell'estensione, senza il punto.

Un'opzione potrebbe essere la divisione dal punto:

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

Nessun errore quando il file non ha estensione:

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

Ma devi stare attento:

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

vale la pena aggiungere un valore inferiore in modo da non chiederti perché i JPG non vengono visualizzati nell'elenco.

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

Tutte le soluzioni sopra funzionano, ma su Linux ho scoperto che c'è una nuova riga alla fine della stringa di estensione che impedirà il successo delle partite. Aggiungi il strip()metodo alla fine. Per esempio:

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

Con splitext ci sono problemi con i file con doppia estensione (ad es. file.tar.gz, Ecc file.tar.bz2.)

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

ma dovrebbe essere: .tar.gz

Le possibili soluzioni sono qui

Puoi trovare alcune cose fantastiche nel modulo pathlib (disponibile in Python 3.x).

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

# Output 
'.txt'

Anche se è un vecchio argomento, ma mi chiedo perché non ci sia nessuno che menzioni un api di python molto semplice chiamato partizione in questo caso:

per ottenere l'estensione di un determinato percorso assoluto del file, puoi semplicemente digitare:

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

esempio:

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

ti darà: 'csv'

Proprio jointutto 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'

Sorpreso, questo non è stato ancora menzionato:

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

Benefici:

• Funziona come previsto per qualsiasi cosa mi venga in mente
• Nessun modulo
• Nessuna regex
• Cross-platform
• Facilmente estensibile (ad es. Nessun punto iniziale per l'estensione, solo l'ultima parte dell'estensione)

Come funzione:

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

Puoi usare a splitsu a filename:

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

Questo non richiede libreria aggiuntiva

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

Questa è una tecnica di rappresentazione diretta delle stringhe: vedo molte soluzioni menzionate, ma penso che la maggior parte stia guardando la divisione. Split lo fa comunque ad ogni occorrenza di "." . Quello che preferiresti cercare è la partizione.

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

Un'altra soluzione con la giusta divisione:

# 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])

Anche a questa domanda è già stata data risposta Aggiungerei la soluzione in Regex.

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

Un vero one-liner, se ti piace regex. E non importa anche se hai "". nel mezzo

import re

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

Vedi qui per il risultato: clicca qui

Questo è il metodo più semplice per ottenere sia il nome file che l'estensione in una sola riga .

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

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

A differenza di altre soluzioni, non è necessario importare alcun pacchetto per questo.

Per i più divertenti ... basta raccogliere le estensioni in un dict e tracciarle tutte in una cartella. Quindi basta estrarre le estensioni desiderate.

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)

prova questo:

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. ottenere tutto il nome del file all'interno dell'elenco
2. dividere il nome del file e controllare la penultima estensione, è nell'elenco pen_ext o no?
3. se sì, uniscilo con l'ultima estensione e impostalo come estensione del file
4. in caso contrario, basta inserire l'ultima estensione come estensione del file
5. e poi dai un'occhiata

# 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(".")

Questo ti darà il nome del file fino al primo ".", Che sarebbe il più comune.