Python 获取扩展名

1. 简介

在编程中，我们经常需要处理文件的名称和路径。为了更好地理解和操作文件，了解文件的扩展名是很有必要的。本文将详细介绍如何使用 Python 获取文件的扩展名。

2. 什么是扩展名

扩展名是文件名的一部分，用来表示文件的类型或格式。它通常位于文件名的最后一个点号后面，如.txt、.jpg、.pdf等。

3. 使用 `os.path` 模块获取扩展名

Python 的内置模块 os.path 提供了一些常用的方法来处理文件路径。其中，os.path.splitext() 方法可以方便地获取文件的扩展名。

下面是使用 os.path.splitext() 方法获取文件扩展名的示例代码：

import os

filename = "example.txt"
extension = os.path.splitext(filename)[1]

print(extension)

代码运行结果：

.txt

在示例代码中，我们首先导入了 os 模块，然后定义了一个文件名 filename。接下来，我们使用 os.path.splitext() 方法获取文件的扩展名，并将其赋值给变量 extension。最后，使用 print() 函数打印出扩展名。

4. 使用 `Path` 类获取扩展名

Python 3.4 引入了 pathlib 模块，提供了更直观和面向对象的方式来处理文件路径。pathlib 模块中的 Path 类可以方便地获取文件的扩展名。

下面是使用 Path 类获取文件扩展名的示例代码：

from pathlib import Path

filename = "example.txt"
extension = Path(filename).suffix

print(extension)

代码运行结果：

.txt

在示例代码中，我们首先导入了 Path 类，然后定义了一个文件名 filename。接下来，我们使用 Path 类将文件名转换为路径对象，并通过 suffix 属性获取扩展名。最后，使用 print() 函数打印出扩展名。

5. 自定义函数获取扩展名

除了使用内置的模块和类，我们还可以自定义函数来获取文件的扩展名。下面是一个简单的自定义函数，它通过逐个字符地查找文件名中的最后一个点号，然后返回点号后面的部分作为扩展名。

def get_extension(filename):
    dot_index = -1

    for i, c in enumerate(reversed(filename)):
        if c == ".":
            dot_index = -i - 1
            break

    return filename[dot_index:]

filename = "example.txt"
extension = get_extension(filename)

print(extension)

代码运行结果：

.txt

在示例代码中，我们定义了一个名为 get_extension() 的函数。首先，我们将 dot_index 初始化为 -1，表示尚未找到点号。然后，使用 enumerate() 函数逐个遍历文件名的字符。当遇到点号时，我们将其索引赋值给 dot_index，并使用 break 关键字跳出循环。最后，使用切片操作符 filename[dot_index:] 返回扩展名。