天天看点

Python:获取文件的扩展名

def get_file_extension(filename):
    arr = os.path.splitext(filename)
    return arr[len(arr) - 1]
    #return arr[len(arr) - 1].replace(".","") 可以将结果中的.去掉


print(get_file_extension("abc.jpg"))
#返回 .jpg

print(get_file_extension("2014.01.25.gif"))
#返回 .gif