Python获取文件扩展名

1. 介绍

在计算机中，文件扩展名是指文件名的最后一个点后面的部分。获取文件扩展名对于文件处理和格式判断是非常重要的。在本文中，我们将详细介绍使用Python语言获取文件扩展名的几种方法。

2. 使用os模块

Python内置的os模块提供了一种获取文件扩展名的简单方法。它提供了一些用于操作文件和目录的函数和方法。

import os

# 获取文件扩展名
def get_file_extension(file_name):
    _, extension = os.path.splitext(file_name)
    return extension

# 测试
file_name = "example.txt"
extension = get_file_extension(file_name)
print(f"文件扩展名为： {extension}")

输出：

文件扩展名为： .txt

在上述示例中，我们使用了os.path.splitext()方法来拆分文件名和扩展名。splitext()方法返回一个元组，其中索引0包含文件名，索引1包含文件扩展名。

3. 使用pathlib模块

从Python 3.4开始，标准库中引入了pathlib模块，为我们提供了处理文件和目录路径的高级接口。

from pathlib import Path

# 获取文件扩展名
def get_file_extension(file_name):
    path = Path(file_name)
    return path.suffix

# 测试
file_name = "example.txt"
extension = get_file_extension(file_name)
print(f"文件扩展名为： {extension}")

输出：

文件扩展名为： .txt

在上述示例中，我们使用了pathlib.Path()创建一个Path对象，并使用suffix属性来获取文件扩展名。

4. 使用split方法

除了使用os和pathlib模块外，Python字符串的split()方法也可以用于获取文件扩展名。

# 获取文件扩展名
def get_file_extension(file_name):
    return file_name.split(".")[-1]

# 测试
file_name = "example.txt"
extension = get_file_extension(file_name)
print(f"文件扩展名为： {extension}")

输出：