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()函数能够更好地处理文件和路径相关的问题。