Sto lavorando a un determinato programma in cui devo fare cose diverse a seconda dell'estensione del file. Potrei semplicemente usare questo?
if m == *.mp3
...
elif m == *.flac
...
Sto lavorando a un determinato programma in cui devo fare cose diverse a seconda dell'estensione del file. Potrei semplicemente usare questo?
if m == *.mp3
...
elif m == *.flac
...
Risposte:
Supponendo che m
sia una stringa, puoi usare endswith
:
if m.endswith('.mp3'):
...
elif m.endswith('.flac'):
...
Per non distinguere tra maiuscole e minuscole e per eliminare una catena altrimenti-if potenzialmente grande:
m.lower().endswith(('.png', '.jpg', '.jpeg'))
.split('.')[-1]
? O la ripartizione è davvero ad alta efficienza?
os.path
fornisce molte funzioni per manipolare percorsi / nomi di file. ( documenti )
os.path.splitext
prende un percorso e divide l'estensione del file dalla fine di esso.
import os
filepaths = ["/folder/soundfile.mp3", "folder1/folder/soundfile.flac"]
for fp in filepaths:
# Split the extension from the path and normalise it to lowercase.
ext = os.path.splitext(fp)[-1].lower()
# Now we can simply use == to check for equality, no need for wildcards.
if ext == ".mp3":
print fp, "is an mp3!"
elif ext == ".flac":
print fp, "is a flac file!"
else:
print fp, "is an unknown file format."
dà:
/folder/soundfile.mp3 è un mp3! folder1 / folder / soundfile.flac è un file flac!
/.mp3
non è considerato un file mp3. Questo è tuttavia il modo in cui uno spazio guida dovrebbe essere trattato. Ad esempio, .gitignore
non è un formato di file
Guarda il modulo fnmatch. Questo farà quello che stai cercando di fare.
import fnmatch
import os
for file in os.listdir('.'):
if fnmatch.fnmatch(file, '*.txt'):
print file
un modo semplice potrebbe essere:
import os
if os.path.splitext(file)[1] == ".mp3":
# do something
os.path.splitext(file)
restituirà una tupla con due valori (il nome file senza estensione + solo l'estensione). Il secondo indice ([1]) fornirà quindi solo l'estensione. Il bello è che in questo modo puoi anche accedere facilmente al nome del file, se necessario!
Un vecchio thread, ma può aiutare i futuri lettori ...
Vorrei evitare di utilizzare .lower () su nomi di file se non altro per rendere il codice più indipendente dalla piattaforma. (Linux fa distinzione tra maiuscole e minuscole, .lower () su un nome file sicuramente corromperà la tua logica alla fine ... o peggio, un file importante!)
Perché non usare re ? (Anche se per essere ancora più robusto, dovresti controllare l'intestazione del file magico di ogni file ... Come controllare il tipo di file senza estensioni in Python? )
import re
def checkext(fname):
if re.search('\.mp3$',fname,flags=re.IGNORECASE):
return('mp3')
if re.search('\.flac$',fname,flags=re.IGNORECASE):
return('flac')
return('skip')
flist = ['myfile.mp3', 'myfile.MP3','myfile.mP3','myfile.mp4','myfile.flack','myfile.FLAC',
'myfile.Mov','myfile.fLaC']
for f in flist:
print "{} ==> {}".format(f,checkext(f))
Produzione:
myfile.mp3 ==> mp3
myfile.MP3 ==> mp3
myfile.mP3 ==> mp3
myfile.mp4 ==> skip
myfile.flack ==> skip
myfile.FLAC ==> flac
myfile.Mov ==> skip
myfile.fLaC ==> flac
import os
source = ['test_sound.flac','ts.mp3']
for files in source:
fileName,fileExtension = os.path.splitext(files)
print fileExtension # Print File Extensions
print fileName # It print file name
if (file.split(".")[1] == "mp3"):
print "its mp3"
elif (file.split(".")[1] == "flac"):
print "its flac"
else:
print "not compat"
.
s, per esempiosome.test.file.mp3
#!/usr/bin/python
import shutil, os
source = ['test_sound.flac','ts.mp3']
for files in source:
fileName,fileExtension = os.path.splitext(files)
if fileExtension==".flac" :
print 'This file is flac file %s' %files
elif fileExtension==".mp3":
print 'This file is mp3 file %s' %files
else:
print 'Format is not valid'
file='test.xlsx'
if file.endswith('.csv'):
print('file is CSV')
elif file.endswith('.xlsx'):
print('file is excel')
else:
print('none of them')