Python 获取文件扩展名

本教程将详细介绍如何使用Python 3获取文件的扩展名。这个过程需要使用Python内置的os.path模块中的splitext()函数。请确保您的Python版本为3.x，因为本教程将基于此版本进行介绍。

1. os.path模块简介

os.path是一个处理文件路径和目录路径的Python模块。它提供了很多有用的函数，其中就包括我们要使用的splitext()函数。

2. splitext()函数简介

os.path.splitext(path)是os.path模块提供的一个方法，它将路径拆分为两部分：根据最后一个点（如果存在），返回文件名和扩展名。

3. 获取文件扩展名

假设我们有以下文件：example.txt, image.jpg, document.docx, 并且它们都位于当前工作目录中。

我们可以通过编写一个简单的Python程序来获取这些文件的扩展名，如下所示：

import os

# 定义文件列表
files = ['example.txt', 'image.jpg', 'document.docx']

for file in files:
    # 使用splitext()函数获取扩展名
    filename, extension = os.path.splitext(file)
    print(f'The file {filename} has the extension {extension}')

在这段代码中，我们首先导入了os.path模块。然后，我们定义了一个文件列表，其中包含一些示例文件名。接下来，我们循环遍历文件列表，对于每个文件，我们使用splitext()函数将文件名和扩展名分开，并打印出结果。

运行这段代码将输出以下内容：

The file example has the extension .txt
The file image has the extension .jpg
The file document has the extension .docx

4. 注意事项

  • splitext()函数返回的扩展名包括点（如.txt, .jpg, .docx）。如果你不想要这个点，可以使用Python字符串切片操作来去掉它。
  • 如果文件没有扩展名（例如，file而不是file.txt），则splitext()函数将返回一个空字符串作为扩展名。
  • splitext()函数只基于最后一个点来分隔文件名和扩展名。如果文件名中包含多个点（例如，example.tar.gz），则可能需要使用其他方法来获取正确的扩展名。