📅  最后修改于: 2023-12-03 15:03:54.077000             🧑  作者: Mango
在编程中,我们往往需要对不同的文件类型进行处理,所以需要识别文件的扩展名。在Python中,识别文件扩展名可以通过以下两种方式实现。
Python标准库中的os模块提供了一种获取文件扩展名的方法,即os.path.splitext()方法。该方法接收一个文件路径作为参数,返回一个包含文件名和扩展名的元组。
下面是一个示例代码,演示了如何使用os.path.splitext()方法获取文件扩展名:
import os
filename = 'example.txt'
name, ext = os.path.splitext(filename)
print(name) # 输出 example
print(ext) # 输出 .txt
Python的标准库中还有另外一个模块叫做mimetypes,该模块可以让我们通过文件名或扩展名获取MIME类型。
下面是一个示例代码,演示了如何使用mimetypes模块获取文件扩展名:
import mimetypes
filename = 'example.txt'
mime_type, encoding = mimetypes.guess_type(filename)
print(mime_type) # 输出 text/plain
print(encoding) # 输出 None
除了使用guess_type()方法,mimetypes模块还提供了其他方法用于获取MIME类型。
以上就是Python中识别文件扩展名的两种实现方式,使用os.path.splitext更简单、原生;而mimetypes则是更灵活、可获取MIME类型。根据实际情况选择即可。