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

在 Python 中获取文件扩展名

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

import os

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

print(file_name)
print(file_extension)

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

输出:

Python中的文件扩展名

• 在第一个示例中,我们直接将元组值解包到两个变量中。

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

• 在第三个例子中,目录名中有一个点。

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

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

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

结论

使用标准方法获取文件扩展名总是更好。如果您已经在使用 os 模块,请使用 splitext() 方法。对于面向对象的方法,使用 pathlib 模块。