python 获取文件名字后缀

原创

mob649e8154b5bf 2024-02-20 03:16:12 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e8154b5bf的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python获取文件名字后缀

在日常的编程过程中，我们经常需要获取文件的后缀名来进行相应的处理，比如判断文件类型、筛选文件等。Python作为一门流行的编程语言，提供了简单的方法来获取文件的后缀名。本文将介绍如何使用Python来获取文件的后缀名，并通过代码示例来加深理解。

什么是文件后缀名？

文件后缀名是文件名中最后一个点（.）之后的部分，用来表示文件的类型。通常情况下，文件后缀名用于区分不同类型的文件，比如.txt表示文本文件，.jpg表示图片文件等。

如何获取文件后缀名？

在Python中，我们可以使用os模块来获取文件的后缀名。os.path.splitext()函数可以将文件名分割成文件名和后缀名两部分，然后我们可以通过取得后缀名部分来获取文件的后缀名。

下面是一个简单的示例代码：

import os

file_name = "example.txt"
file_ext = os.path.splitext(file_name)[1]

print("文件后缀名为：" + file_ext)

在这段代码中，我们首先导入os模块，然后定义一个文件名example.txt。通过调用os.path.splitext()函数，我们可以获取到文件的后缀名，并将其打印输出。

代码示例

下面我们通过一个更加完整的代码示例来演示如何获取一个文件夹中所有文件的后缀名：

import os

def get_file_extension(file_path):
    return os.path.splitext(file_path)[1]

def get_all_file_extensions(folder_path):
    file_extensions = []
    for file_name in os.listdir(folder_path):
        file_path = os.path.join(folder_path, file_name)
        if os.path.isfile(file_path):
            file_ext = get_file_extension(file_name)
            file_extensions.append(file_ext)
    return file_extensions

folder_path = "./files/"
extensions = get_all_file_extensions(folder_path)

print("文件夹中所有文件的后缀名为：", extensions)

在这段代码中，我们定义了两个函数get_file_extension()和get_all_file_extensions()，分别用于获取单个文件的后缀名和获取文件夹中所有文件的后缀名。通过遍历文件夹中的所有文件，并调用get_file_extension()函数来获取文件的后缀名，最终将所有文件的后缀名保存在一个列表中并打印输出。