Python实用技巧: 获取 后缀名(扩展名) 或 文件名

在Python中，你可以使用内置的os.path模块来获取文件的扩展名（后缀名）或者完整的文件名。以下是两个函数，分别用于获取文件的扩展名和文件名。

import os
 
def get_file_extension(filename):
    return os.path.splitext(filename)[-1]
 
def get_file_name(filename):
    return os.path.basename(filename)
 
# 示例
filename = 'example.txt'
extension = get_file_extension(filename)
print(f"The extension is: {extension}")  # 输出: The extension is: .txt
 
filename_only = get_file_name(filename)
print(f"The file name is: {filename_only}")  # 输出: The file name is: example.txt

这段代码首先导入了os模块，然后定义了两个函数get_file_extension和get_file_name。get_file_extension使用os.path.splitext()函数分离文件名和扩展名，返回扩展名。get_file_name使用os.path.basename()函数返回文件名，不包含路径。

在示例中，我们使用了文件名example.txt，并打印出了它的扩展名和文件名。