我在一个特定的程序上工作,我需要根据文件的扩展名做不同的事情。我能用这个吗?
if m == *.mp3
...
elif m == *.flac
...
我在一个特定的程序上工作,我需要根据文件的扩展名做不同的事情。我能用这个吗?
if m == *.mp3
...
elif m == *.flac
...
当前回答
查看模块fnmatch。它会达到你想要的效果。
import fnmatch
import os
for file in os.listdir('.'):
if fnmatch.fnmatch(file, '*.txt'):
print file
其他回答
假设m是一个字符串,你可以使用endswith:
if m.endswith('.mp3'):
...
elif m.endswith('.flac'):
...
不区分大小写,并消除可能较大的else-if链:
m.lower().endswith(('.png', '.jpg', '.jpeg'))
或者:
from glob import glob
...
for files in glob('path/*.mp3'):
do something
for files in glob('path/*.flac'):
do something else
#!/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'
从Python3.4开始使用。
from pathlib import Path
Path('my_file.mp3').suffix == '.mp3'
如果您正在处理包含句点的文件夹,则可以使用
Path('your_folder.mp3').is_file() and Path('your_folder.mp3').suffix == '.mp3'
确保后缀为.mp3的文件夹不会被解释为mp3文件。
一个简单的方法是:
import os
if os.path.splitext(file)[1] == ".mp3":
# do something
Os.path.splitext (file)将返回一个包含两个值的元组(没有扩展名的文件名+只有扩展名的文件名)。因此,第二个索引([1])只提供扩展名。最酷的是,如果需要的话,这样你也可以很容易地访问文件名!