Python os.path.splitext函数的用法及示例

一、什么是os.path.splitext函数

os.path模块是Python标准库中用于处理文件路径的模块，其中os.path.splitext函数是用于将文件路径分割成文件名和扩展名的函数。

    import os
    path = '/path/to/file.txt'
    filename, ext = os.path.splitext(path)
    print(filename)  # /path/to/file
    print(ext)  # .txt

上述代码将文件路径 '/path/to/file.txt' 分割成文件名 '/path/to/file' 和扩展名 '.txt'。

二、os.path.splitext函数的参数

os.path.splitext函数只有一个参数，即需要分割的文件路径。

三、os.path.splitext函数的返回值

os.path.splitext函数的返回值是一个元组，其中包含分割后的文件名和扩展名。

    import os
    path = '/path/to/file.txt'
    filename, ext = os.path.splitext(path)
    print(filename)  # /path/to/file
    print(ext)  # .txt

四、os.path.splitext函数的使用场景

1. 获取文件扩展名

os.path.splitext函数最常见的用途是获取文件的扩展名。

    import os
    path = '/path/to/file.txt'
    _, ext = os.path.splitext(path)
    print(ext)  # .txt

上述代码中，由于不需要使用文件名，所以将其赋值给下划线变量 _。

2. 更改文件扩展名

os.path.splitext函数也可以用于更改文件的扩展名。

    import os
    path = '/path/to/file.txt'
    new_path = os.path.splitext(path)[0] + '.jpg'
    print(new_path)  # /path/to/file.jpg

上述代码中，将原文件的扩展名 .txt 更改为 .jpg 。

3. 文件名去除扩展名

os.path.splitext函数还可以把文件名去除掉扩展名。

    import os
    path = '/path/to/file.txt'
    filename = os.path.splitext(os.path.basename(path))[0]
    print(filename)  # file

上述代码中，使用os.path.basename函数获取文件名 'file.txt'，然后再使用os.path.splitext函数将其分割成 'file' 和 '.txt'，最后取 'file' 作为文件名。

五、总结

os.path.splitext函数可以方便地将文件路径分割成文件名和扩展名，常见的使用场景包括获取文件扩展名、更改文件扩展名、文件名去除扩展名等。其返回值是一个包含文件名和扩展名的元组，可以通过下标或多重赋值的方式获取。