天天看點

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