python 获取文件后缀

# 获取文件后缀的示例代码

# 方法一：使用 os.path.splitext()
import os

def get_file_extension(file_path):
    # 使用 os.path.splitext() 分割文件名和扩展名，返回一个元组 (filename, extension)
    return os.path.splitext(file_path)[1]

# 示例
file_path = "example.txt"
extension = get_file_extension(file_path)
print(f"文件 {file_path} 的后缀是: {extension}")

# 方法二：使用 pathlib.Path.suffix
from pathlib import Path

def get_file_extension_pathlib(file_path):
    # 使用 pathlib.Path 对象的 suffix 属性获取文件后缀
    return Path(file_path).suffix

# 示例
file_path = "example.txt"
extension = get_file_extension_pathlib(file_path)
print(f"文件 {file_path} 的后缀是: {extension}")

解释说明：

1. 方法一：使用 os.path.splitext() 函数来分割文件路径，返回一个包含文件名和扩展名的元组。[1] 表示取元组中的第二个元素，即文件扩展名。
2. 方法二：使用 pathlib.Path 模块的 suffix 属性直接获取文件的扩展名。Path 对象提供了更现代和面向对象的方式来处理文件路径。

这两种方法都可以有效地获取文件的后缀名。