如何使用Python提取文件扩展名？

在编程过程中，有时候需要获取文件的扩展名，以便后续的操作。Python语言提供了多种方法来实现这一功能。下面我们就来分别介绍一下。

阅读更多：Python 教程

方法一：使用split()函数

用Python内置函数split()可以通过分离文件路径和文件扩展名来获取文件扩展名。示例代码如下所示：

filename = "hello.py"
extension = filename.split(".")[-1]
print(extension)

运行结果如下：

py

这段代码将字符串 “hello.py” 分割成一个包含两个元素的列表 [“hello”, “py”]，并返回最后一个元素 “py”。这种方法在处理只有一个点的文件名时也是可行的。

方法二：使用os.path.splittext()函数

Python内置模块os中提供了一个实用函数splittext()，可以帮助我们从文件名中准确地获取文件扩展名。示例代码如下所示：

import os

filename = "hello.py"
extension = os.path.splitext(filename)[-1]
print(extension)

运行结果如下：

.py

这段代码将传递给splitext()函数的文件名“hello.py”分离为一个路径和扩展名部分。[-1]标志着代码只返回最后一个元素，也就是获取文件扩展名“.py”。

方法三：使用Python的正则表达式re

此方法使用Python中的正则表达式模块re来搜索字符串中的文件扩展名。示例代码如下所示：

import re

filename = "hello.py"
extension = re.findall("\.[^.]*$",filename)
print(extension[0])

运行结果如下：

.py

使用正则表达式可以更好地控制文件名的正确性，并确保得到正确的文件扩展名。上述代码中的正则表达式“.[^.]*$”将匹配一个点前面的任何字符和点后面的任何字符，在此处匹配”.py”。

方法四：使用Pathlib模块

Python3.4 中引入了Pathlib模块，它包含了所有处理路径的常见工具。示例代码如下所示：

from pathlib import Path

filename = "hello.py"
extension = Path(filename).suffix
print(extension)

运行结果如下：

.py

这段代码使用Python路径库（Pathlib）中的函数suffix，它直接返回文件扩展名“.py”，而不是文件名“hello.py”。

结论

在Python中，我们可以选择多种方法获取文件的扩展名。无论您最终选择哪种方法，都可以便捷地从文件名中提取扩展名用于开发或处理。