Python中如何获取文件的扩展名

在Python中获取文件的扩展名可以通过多种方式实现。下面将介绍几种常见的方法，并给出相应的使用例子。

1. 使用split()方法：

split()方法可以将字符串按照指定的分隔符进行切割，并返回切割后的子字符串列表。我们可以将文件路径或文件名作为输入，使用split('.')将其按照点号（.）进行切割，取得最后一个元素即为文件的扩展名。

file_name = "test.txt"
extension = file_name.split('.')[-1]
print(extension)  # 输出：txt

file_path = "/path/to/file/test.txt"
extension = file_path.split('.')[-1]
print(extension)  # 输出：txt

2. 使用os模块：

os模块是Python标准库中提供的一个与操作系统交互的接口。os.path模块中的splitext()方法可以将文件名与扩展名分开，并以元组的形式返回。我们可以通过解析返回的元组获取文件的扩展名。

import os

file_name = "test.txt"
extension = os.path.splitext(file_name)[1][1:]
print(extension)  # 输出：txt

file_path = "/path/to/file/test.txt"
extension = os.path.splitext(file_path)[1][1:]
print(extension)  # 输出：txt

3. 使用re模块：

re模块是Python中用于执行正则表达式操作的库。我们可以使用re模块中的search()方法从文件路径中提取扩展名。通过传入一个正则表达式模式来匹配文件路径中的扩展名部分。

import re

file_name = "test.txt"
extension = re.search("\.([a-zA-Z0-9]+)$", file_name).group(1)
print(extension)  # 输出：txt

file_path = "/path/to/file/test.txt"
extension = re.search("\.([a-zA-Z0-9]+)$", file_path).group(1)
print(extension)  # 输出：txt

以上是获取文件扩展名的三种常见方法，可以根据实际情况选择合适的方法来获取文件的扩展名。