Python os.path.splitext：分離文件名和擴展名

一、介紹

Python中的os.path模塊提供了很多處理文件和目錄的方法，其中os.path.splitext()方法用於分離文件名和擴展名。

在文件處理的過程中，我們常常需要對文件名和擴展名進行分離處理，os.path.splitext()方法能夠簡單快速地幫助我們實現這個功能。

二、用法

os.path.splitext(path)方法接受一個字符串作為參數，該字符串表示文件路徑，返回一個包含文件路徑和擴展名的元組。

如果路徑參數中沒有擴展名，則返回的元組中擴展名部分為空字符串。

import os

file_path = '/path/to/file.txt'
filename, ext = os.path.splitext(file_path)

print('File name:', filename)
print('Extension:', ext)

該代碼的輸出結果為：

File name: /path/to/file
Extension: .txt

注意：os.path.splitext()方法並不檢查文件是否存在，它只是處理一個文件路徑字符串。

三、示例

1. 處理多個文件

os.path.splitext()方法可以很方便地對多個文件進行批量處理。

import os

files = ['/path/to/file1.txt', '/path/to/file2.jpg', '/path/to/file3.py']

for file in files:
    filename, ext = os.path.splitext(file)
    print('File name:', filename)
    print('Extension:', ext)

該代碼的輸出結果為：

File name: /path/to/file1
Extension: .txt
File name: /path/to/file2
Extension: .jpg
File name: /path/to/file3
Extension: .py

2. 只列出指定擴展名的文件

在打印指定目錄下所有文件時，可以使用os.path.splitext()方法來只列出指定擴展名的文件。

import os

dir_path = '/path/to/dir'
for file in os.listdir(dir_path):
    if os.path.splitext(file)[-1] == '.txt':
        print(file)

該代碼只列出dir_path目錄下所有擴展名為.txt的文件。

3. 創建不帶擴展名的文件

在創建文件時，os.path.splitext()方法可以很容易地幫助我們創建不帶擴展名的文件。

import os

dir_path = '/path/to/dir'
filename = 'myfile'
ext = '.txt'
file_path = os.path.join(dir_path, filename+ext)

with open(file_path, 'w') as f:
    f.write('Hello World!')

該代碼創建一個名為myfile.txt的空文件，並向其中寫入Hello World!。

四、總結

os.path.splitext()方法可以快捷地對文件名和擴展名進行分離處理，適用於各種文件處理場景。

它還可以與其他os.path模塊中的方法配合使用，比如os.path.join()方法可以創建一個文件路徑字符串。