Python 在Python中提取文件名的扩展名

在本文中，我们将介绍如何在Python中提取文件名的扩展名。扩展名是指文件名称中最后一个点之后的字符串，通常用于确定文件的类型或格式。Python提供了多种方法来提取文件名的扩展名，我们将逐一介绍这些方法。

方法一：使用split()函数

Python的字符串有一个split()函数，可以接受一个参数作为分隔符，并返回一个包含分隔后的子字符串的列表。我们可以使用split()函数来提取文件名的扩展名。

下面是一个使用split()函数提取文件名扩展名的示例：

filename = "example.txt"
extension = filename.split(".")[-1]
print(extension)  # 输出：txt

首先，我们将文件名赋值给变量filename。然后，使用split(“.”)将文件名拆分成一个字符串列表。由于扩展名通常位于列表的最后一个元素，我们可以使用[-1]索引来访问该元素，并将其赋值给变量extension。最后，打印出extension的值，即可得到文件名的扩展名。

需要注意的是，这种方法仅适用于文件名中只有一个点的情况。如果文件名中包含多个点，且我们只对最后一个点之后的字符串感兴趣，那么这种方法就不适用了。

方法二：使用os模块

Python的内置模块os提供了一系列用于处理文件和目录的函数。其中的os.path模块提供了一些有用的函数来处理文件路径和名称。

我们可以使用os.path模块的函数来提取文件名的扩展名。下面是一个使用os.path模块提取文件名扩展名的示例：

import os

filename = "example.txt"
extension = os.path.splitext(filename)[1]
print(extension)  # 输出：.txt

首先，我们导入了os模块。然后，将文件名赋值给变量filename。使用os.path模块的splitext()函数，可以将文件名拆分为文件的名称部分和扩展名部分，并以列表形式返回。我们只需要使用[1]索引来获取扩展名部分，并将其赋值给变量extension。最后，打印出extension的值，即可得到文件名的扩展名。

这种方法适用于文件名中包含多个点的情况，并且只对最后一个点之后的字符串感兴趣。

方法三：使用正则表达式

正则表达式是一种强大的模式匹配工具，可以用于在字符串中查找和提取特定的模式。我们可以使用正则表达式来提取文件名的扩展名。

下面是一个使用正则表达式提取文件名扩展名的示例：

import re

filename = "example.txt"
extension = re.findall(r"\.(\w+)$", filename)[0]
print(extension)  # 输出：txt

首先，我们导入了re模块。然后，将文件名赋值给变量filename。使用re模块的findall()函数，结合正则表达式"\.(\w+)$"，可以在文件名中查找并返回匹配的扩展名。正则表达式中的\.(\w+)$表示以点开头，后面跟着一个或多个字母、数字或下划线，并且以字符串的结尾结束。使用[0]索引可以获取匹配的扩展名部分，并将其赋值给变量extension。最后，打印出extension的值，即可得到文件名的扩展名。

需要注意的是，如果文件名中没有找到匹配的扩展名，findall()函数将返回一个空列表。为了避免出现IndexError异常，我们可以在访问列表元素之前，先使用len()函数检查列表的长度是否大于0。