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()函数将文件重命名为新的文件名和扩展名组合而成的字符串。