Python 程序：从文件名提取扩展名

要理解这个例子，你应该具备以下Python编程主题的知识：

• Python目录和文件管理
• Python文件I/O操作

示例 1：使用os模块的splitext()方法

import os
file_details = os.path.splitext('/path/file.ext')
print(file_details)
print(file_details[1])

输出

('/path/file', '.ext')
.ext

os.path.splitext()返回一个元组，一个元素是包含路径的文件名，另一个是文件的扩展名。如果你只想获取文件扩展名，可以像上面所示打印file_details[1]。

示例 2：使用pathlib模块

import pathlib
print(pathlib.Path('/path/file.ext').suffix)

输出

.ext

使用pathlib模块的suffix属性，我们可以获取文件的扩展名。在上面的例子中，.ext是文件file.ext的扩展名。

注意：这适用于Python 3.4及以上版本。