如何获取文件后缀名 获取文件后缀名的方法

如何获取文件后缀名

文件后缀名是文件名中最后一个 '.' 之后的字符串，它能够指示文件的类型。例如，文件名为example.png，后缀名为.png，说明这是一张图片文件。获取文件的后缀名在编程中经常用到，下面我们就来讨论一下如何获取文件后缀名。

1、使用字符串切割

文件后缀名通常是通过字符串切割来获取的。假设我们有一个文件的路径字符串，我们可以将该字符串按照 '.' 进行切割，然后获取最后一个元素作为文件后缀名。

filePath = "example.png"
suffix = filePath.split(".")[-1]
print(suffix)

以上程序的输出结果为：

png

以上方法只适用于文件名只有一个 '.' 的情况，如果文件名中存在多个 '.'，则需要对字符串进行更细致的切割。

2、使用os模块获取文件后缀名

Python的os模块提供了访问操作系统功能的接口，可以获取文件的路径、大小、创建时间等信息，同时也可以轻松获取文件的后缀名。

import os
filePath = "example.png"
suffix = os.path.splitext(filePath)[-1]
print(suffix)

以上程序的输出结果为：

.png

上述程序中，os.path.splitext() 可以将文件路径拆分为文件名和文件后缀名两个部分，其中后缀名包含 '.'，如果不需要 '.'，可以在获取到后缀名之后使用字符串切片去掉。

3、使用re模块获取文件后缀名

使用正则表达式也可以获取文件后缀名，正则表达式的特点是能够处理比较复杂的字符串匹配，因此可以适用于文件名中存在多个 '.' 的情况。

import re
filePath = "example.ac.fg.png"
suffix = re.findall("\.([^\.]*$)", filePath)[-1]
print(suffix)

以上程序的输出结果为：

png

以上程序使用正则表达式 \.([^\.]*$) 获取文件后缀名，\ 表示转义字符，\. 匹配 '.'，() 表示匹配任意字符，[^\.] 除了 '.' 外的所有字符，* 表示匹配任意长度的字符，$ 表示行尾。re.findall() 方法可以返回所有匹配的字符串，需要取最后一个字符串作为文件后缀名。

4、使用第三方库获取文件后缀名

除了使用Python自带的库外，还可以使用第三方库来获取文件的后缀名，比如mimetypes库、magic库等。

import mimetypes
filePath = "example.png"
suffix = mimetypes.guess_extension(filePath)
print(suffix)

以上程序的输出结果为：

.png

mimetypes库可以根据文件名获取文件的MIME类型，然后返回与该MIME类型相关的文件拓展名，根据文件拓展名即可获取文件后缀名。

结论

以上就是获取文件后缀名的常用方法，其中最常用的方法是使用os模块的os.path.splitext()函数。要根据具体情况选择合适的方法，避免出现错误。