一、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()函数能够更好地处理文件和路径相关的问题。
原创文章，作者：UYLR，如若转载，请注明出处：https://www.506064.com/n/146367.html
微信扫一扫 
支付宝扫一扫 