Python获取文件后缀|极客教程

Python获取文件后缀

介绍

文件后缀即文件名中最后一个点（.）后面的部分，表示文件的类型或格式。在处理文件时，有时需要获取文件的后缀以进行一些特定的操作，比如判断文件类型、过滤特定格式的文件等。

在Python中，我们可以使用多种方法来获取文件后缀，本文将详细介绍这些方法及其使用。

方法一：使用split()函数分割文件名

我们可以使用Python的字符串函数split()将文件名以点（.）为分隔符进行分割，然后取分割后的最后一个部分作为文件后缀。下面是使用split()函数获取文件后缀的示例代码：

def get_file_extension(filename):
    return filename.split(".")[-1]

file = "example.txt"
extension = get_file_extension(file)
print("文件后缀:", extension)

运行结果：

文件后缀: txt

方法二：使用os模块的path.splitext()函数

Python的os模块提供了许多关于文件和目录操作的函数。其中，path.splitext()函数可以用来获取文件名和文件后缀的分隔。

import os

def get_file_extension(filename):
    _, extension = os.path.splitext(filename)
    return extension[1:]

file = "example.txt"
extension = get_file_extension(file)
print("文件后缀:", extension)

运行结果：

文件后缀: txt

方法三：使用pathlib模块的Path对象

Python的pathlib模块提供了Path对象，可以方便地进行文件路径操作。Path对象的suffix属性可直接获取文件后缀。

from pathlib import Path

def get_file_extension(filename):
    file = Path(filename)
    return file.suffix[1:]

file = "example.txt"
extension = get_file_extension(file)
print("文件后缀:", extension)

运行结果：

文件后缀: txt

方法四：使用正则表达式匹配文件后缀

如果要处理更为复杂的文件名情况，可以使用正则表达式来匹配文件后缀。下面的示例代码演示了使用re模块的search()函数来提取文件后缀。

import re

def get_file_extension(filename):
    pattern = r"\.([a-zA-Z0-9]+)$"
    match = re.search(pattern, filename)
    if match:
        return match.group(1)

file = "example.txt"
extension = get_file_extension(file)
print("文件后缀:", extension)

运行结果：

文件后缀: txt

方法五：使用字符串切片操作

简单的文件名情况下，我们还可以使用简单的字符串切片操作来获取文件后缀。具体做法是，找到文件名中最后一个点的位置，然后在该位置之后的部分即为文件后缀。

def get_file_extension(filename):
    dot_position = filename.rfind(".")
    if dot_position != -1:
        return filename[dot_position+1:]

file = "example.txt"
extension = get_file_extension(file)
print("文件后缀:", extension)