从文件名中提取扩展名的Python程序

在这个例子中，你将学习如何从文件名中提取扩展名。

例子 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模块的后缀属性，我们可以得到一个文件的扩展名。在上面的例子中，.ext是文件file.ext的扩展名。

注意：它适用于 python 3.4 及以上版本。