我们可以使用Pythonos moduleSplitext()函数来获取文件扩展名。此函数将文件路径拆分为具有两个值的元组-根和扩展名。
在Python中获取文件扩展名
下面是一个简单的程序，可以用来获取Python中的文件扩展名。
1import os
2
3# unpacking the tuple
4file_name, file_extension = os.path.splitext("/Users/pankaj/abc.txt")
5
6print(file_name)
7print(file_extension)
8
9print(os.path.splitext("/Users/pankaj/.bashrc"))
10print(os.path.splitext("/Users/pankaj/a.b/image.png"))
输出 ：
- 在第一个示例中，我们直接将元组值解压缩为两个变量。
- 请注意，.bashrc文件没有扩展名。在文件名中添加圆点以使其成为隐藏文件。
- 在第三个示例中，目录名称中有一个点。
使用Pathlib模块获取文件扩展名
我们也可以使用pathlib模块来获取文件扩展名。该模块是在Python3.4版本中引入的。
1>>> import pathlib
2>>> pathlib.Path("/Users/pankaj/abc.txt").suffix
3'.txt'
4>>> pathlib.Path("/Users/pankaj/.bashrc").suffix
5''
6>>> pathlib.Path("/Users/pankaj/.bashrc")
7PosixPath('/Users/pankaj/.bashrc')
8>>> pathlib.Path("/Users/pankaj/a.b/abc.jpg").suffix
9'.jpg'
10>>>
结论
使用标准方法获取文件扩展名总是更好。如果您已经在使用os模块，则使用plitext()方法。对于面向对象的方法，请使用pathlib模块。