使用pathlib获取文件扩展名

在Python中，使用pathlib模块可以轻松地操作文件路径和文件系统操作。其中一个常见的操作就是获取文件的扩展名。

1. 导入模块

from pathlib import Path

2. 创建Path对象

创建Path对象来表示要操作的文件路径。这个路径可以是相对的或绝对的。

path = Path("example.txt")

3. 获取扩展名

使用suffix属性获取文件的扩展名。

ext = path.suffix

完整代码

from pathlib import Path

path = Path("example.txt")
ext = path.suffix

print(ext)

输出为：

.txt

应用

下面是一个用于遍历目录中的所有文件，并获取其扩展名的示例代码。

from pathlib import Path

def get_ext(path):
    """
    获取文件扩展名
    """
    return path.suffix

def traverse_dir(dir_path):
    """
    遍历目录中的所有文件，并返回其扩展名
    """
    for file_path in dir_path.glob('*'):
        if file_path.is_file():
            ext = get_ext(file_path)
            print(f"{file_path.name}: {ext}")

if __name__ == '__main__':
    dir_path = Path('example_dir')
    traverse_dir(dir_path)

执行结果：

example.txt: .txt
example2.py: .py