Python os.path.splitext – 獲取文件名和擴展名

一、什麼是os.path.splitext

import os
file_path = "/home/user/file.txt"
file_name_with_extension = os.path.basename(file_path)
file_name, file_extension = os.path.splitext(file_name_with_extension)
print(file_name) # file
print(file_extension) # .txt

在Python編程中，處理文件名和文件擴展名是一個很常見的事情。os.path.splitext()是Python的os.path模塊中的一個函數，它能夠方便地將文件名和文件擴展名分開。

os.path.splitext()函數接受一個文件路徑作為參數，並返回一個元組，其中包含文件名和文件擴展名。如果文件沒有擴展名，則文件名和擴展名都將是空字元串。

二、為什麼要使用os.path.splitext

通過使用os.path.splitext()函數，可以在Python程序中輕鬆獲取文件的擴展名。而且，此函數返回的是一個元組，使得在程序中處理文件名和擴展名變得更加方便。

另外，使用os.path.splitext()函數可以避免手動解析文件名和擴展名後綴，從而減少代碼的錯誤和維護成本。

三、如何使用os.path.splitext

os.path.splitext()函數非常簡單易用。只需要將文件的完整路徑作為參數傳遞給函數，並將返回的結果分別存儲到適當的變數中即可。

import os
file_path = "/home/user/file.txt"
file_name_with_extension = os.path.basename(file_path)
file_name, file_extension = os.path.splitext(file_name_with_extension)
print(file_name) # file
print(file_extension) # .txt

在上面的示例中，我們使用os.path.splitext()函數來從給定的文件路徑中提取文件名和擴展名。

首先，我們使用os.path.basename()函數獲取文件名和擴展名。該函數從完整路徑中提取文件名和擴展名，並返回一個包含這兩個值的字元串。

其次，將返回的值傳遞給os.path.splitext()函數，該函數將返回一個元組，其中包含文件名和擴展名。這兩個值分別存儲在file_name和file_extension這兩個變數中。

四、os.path.splitext的應用場景

os.path.splitext()函數可以用在多個場景中：

1. 驗證文件擴展名

數據驗證是編程中不可避免的過程。使用os.path.splitext(),可以驗證一個文件是否具有正確的擴展名。

import os
def is_valid_file(file_path, valid_extensions):
    """
    Check if a file has a valid extension
    """
    file_name_with_extension = os.path.basename(file_path)
    _, file_extension = os.path.splitext(file_name_with_extension)
    return file_extension in valid_extensions

valid_extensions = ['.png', '.jpg', '.jpeg', '.gif']
file_path_1 = '/home/user/image.png'
file_path_2 = '/home/user/document.docx'
print(is_valid_file(file_path_1, valid_extensions)) # True
print(is_valid_file(file_path_2, valid_extensions)) # False

2. 更改文件擴展名

使用os.path.splitext()函數，可以輕鬆地更改文件的擴展名。

import os
def change_file_extension(file_path, new_extension):
    """
    Change the extension of a given file
    """
    file_name_with_extension = os.path.basename(file_path)
    file_name, _ = os.path.splitext(file_name_with_extension)
    new_file_name_with_extension = file_name + new_extension
    new_file_path = os.path.join(os.path.dirname(file_path), new_file_name_with_extension)
    os.rename(file_path, new_file_path)
    return new_file_path

file_path = '/home/user/file.txt'
new_extension = '.docx'
new_file_path = change_file_extension(file_path, new_extension)

在上面的示例中，我們使用os.path.splitext()函數從文件路徑中提取文件名並將其更改為新擴展名。然後，使用os.rename()函數將文件重命名為新的文件名和擴展名組合而成的字元串。