if (file.split(".")[1] == "mp3"):
    print "its mp3"
elif (file.split(".")[1] == "flac"):
    print "its flac"
else:
    print "not compat"

#!/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'

一个简单的方法是:

import os

if os.path.splitext(file)[1] == ".mp3":
    # do something

Os.path.splitext (file)将返回一个包含两个值的元组(没有扩展名的文件名+只有扩展名的文件名)。因此，第二个索引([1])只提供扩展名。最酷的是，如果需要的话，这样你也可以很容易地访问文件名!

假设m是一个字符串，你可以使用endswith:

if m.endswith('.mp3'):
...
elif m.endswith('.flac'):
...

不区分大小写，并消除可能较大的else-if链:

m.lower().endswith(('.png', '.jpg', '.jpeg'))

file='test.xlsx'
if file.endswith('.csv'):
    print('file is CSV')
elif file.endswith('.xlsx'):
    print('file is excel')
else:
    print('none of them')

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