Python os.path splitext函數：獲取文件擴展名

一、splitext函數介紹

Python的os.path模塊提供了很多處理文件和路徑相關的函數。其中，os.path.splitext()函數用於分離文件名和擴展名，返回一個二元組（filename, extension）。

import os.path

filename, extension = os.path.splitext('/path/to/myfile.txt')
print(filename)     # /path/to/myfile
print(extension)    # .txt

在上面的代碼中，我們傳入一個文件路徑，os.path.splitext()函數返回一個二元組，其中第一個元素是文件路徑去除擴展名的部分，第二個元素則是文件的擴展名。

二、獲取文件擴展名

使用os.path.splitext()函數獲取文件的擴展名很簡單，只需要將文件路徑傳入到函數中即可：

import os.path

filename = '/path/to/myfile.txt'
extension = os.path.splitext(filename)[1]
print(extension)    # .txt

上面的代碼中，我們使用os.path.splitext()函數獲取文件的擴展名。函數返回一個二元組，我們取得返回值的第二個元素並賦值給extension變數，最終列印出文件的擴展名。

三、判斷文件類型

在實際的開發中，我們有時需要根據文件擴展名判斷文件類型，比如判斷文件是否為圖片、音頻、視頻、文本等。這時，我們可以使用os.path.splitext函數獲取文件的擴展名，再根據擴展名判斷文件類型。

import os.path

def get_file_type(filename):
    extension = os.path.splitext(filename)[1]
    if extension == '.txt':
        return '文本文件'
    elif extension in ['.jpg', '.png', '.gif']:
        return '圖片文件'
    elif extension in ['.mp3', '.wav', '.ogg']:
        return '音頻文件'
    elif extension in ['.mp4', '.avi', '.mov']:
        return '視頻文件'
    else:
        return '未知文件類型'

print(get_file_type('/path/to/myfile.txt'))    # 文本文件
print(get_file_type('/path/to/myimage.jpg'))   # 圖片文件
print(get_file_type('/path/to/mysong.mp3'))    # 音頻文件
print(get_file_type('/path/to/myvideo.mp4'))   # 視頻文件
print(get_file_type('/path/to/myfile.xyz'))    # 未知文件類型

上面的代碼中，我們定義了一個get_file_type()函數，接收一個文件路徑作為參數。函數使用os.path.splitext()函數獲取文件的擴展名，再根據擴展名返迴文件類型。get_file_type()函數返回的是字元串類型的文件類型，包括文本文件、圖片文件、音頻文件、視頻文件和未知文件類型。

四、小結

Python的os.path模塊提供了很多處理文件和路徑相關的函數。其中，os.path.splitext()函數用於分離文件名和擴展名，返回一個二元組。我們可以使用os.path.splitext()函數獲取文件的擴展名，再根據擴展名判斷文件類型。

在實際的開發中，文件的擴展名常常用於判斷文件類型、文件名的提取等，掌握os.path.splitext()函數能夠更好地處理文件和路徑相關的問題。