我们可以使用Python os模块splitext()函数来获取文件扩展名。
此函数将文件路径拆分为具有两个值的元组-root和extension。

在Python中获取文件扩展名

这是一个使用Python获取文件扩展名的简单程序。

import os

# unpacking the tuple
file_name, file_extension = os.path.splitext("/Users/hyman/abc.txt")

print(file_name)
print(file_extension)

print(os.path.splitext("/Users/hyman/.bashrc"))
print(os.path.splitext("/Users/hyman/a.b/image.png"))

输出：

Python中的文件扩展名

• 在第一个示例中，我们直接将元组值解压缩为两个变量。

• 请注意，.bashrc文件没有扩展名。
  点被添加到文件名以使其成为隐藏文件。

• 在第三个示例中，目录名称中有一个点。

使用Pathlib模块获取文件扩展名

我们还可以使用pathlib模块获取文件扩展名。
该模块在Python 3.4版本中引入。

>>> import pathlib
>>> pathlib.Path("/Users/hyman/abc.txt").suffix
'.txt'
>>> pathlib.Path("/Users/hyman/.bashrc").suffix
''
>>> pathlib.Path("/Users/hyman/.bashrc")
PosixPath('/Users/hyman/.bashrc')
>>> pathlib.Path("/Users/hyman/a.b/abc.jpg").suffix
'.jpg'
>>>