如何使用Python函数获取文件名的扩展名?

要获取文件名的扩展名，可以使用Python的字符串处理方法和函数来实现。下面是一种简单的方法来获取文件名的扩展名：

1. 使用os.path模块的splitext()函数来获取文件名和扩展名的元组。splitext()函数将文件名分割成两部分，返回一个元组，其中 个元素是文件名，第二个元素是扩展名。

import os

file_name = 'example.txt'
file_name_without_ext, file_ext = os.path.splitext(file_name)
print(file_ext)  # 输出: .txt

在这个示例中，我们使用os.path.splitext()函数将文件名拆分为文件名(example)和扩展名(.txt)，并将扩展名存储到变量file_ext中。

2. 使用字符串的split()方法和列表索引来获取扩展名。split()方法将字符串根据指定的分隔符进行分割，并返回一个包含分割后的子字符串的列表。我们可以使用列表索引来获取扩展名。

file_name = 'example.txt'
file_ext = file_name.split('.')[-1]
print(file_ext)  # 输出: txt

在这个示例中，我们使用split('.')将文件名拆分为一个列表，其中 个元素是文件名，第二个元素是扩展名。通过使用列表索引[-1]，我们可以获取列表中的最后一个元素，即扩展名。

3. 使用正则表达式来匹配文件名的扩展名。使用re模块的search()方法可以在字符串中搜索特定的模式。我们可以使用正则表达式来匹配扩展名的模式，并提取出扩展名。

import re

file_name = 'example.txt'
file_ext = re.search(r'\.(\w+)$', file_name).group(1)
print(file_ext)  # 输出: txt

在这个示例中，我们使用正则表达式'\.(\w+)$'来匹配文件名中最后一个点(.)后面的所有字符。使用search()方法搜索文件名，并使用group(1)函数提取匹配的结果。

这些方法提供了不同的途径来获取文件名的扩展名。选择其中一种方法来使用，取决于你的需求和偏好。