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

Python os模块splitext（）

splitext（）函数将文件路径拆分为具有两个值的元组：root和extension。

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

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

获取文件扩展名的Pathlib模块

import pathlib
pathlib.Path("/Users/pankaj/abc.txt").suffix
#output:'.txt'

import os.path
extension = os.path.splitext(filename)[1]

对于简单的用例，一个选项可能是从点拆分：

>>> filename = "example.jpeg"
>>> filename.split(".")[-1]
'jpeg'

文件没有扩展名时没有错误：

>>> "filename".split(".")[-1]
'filename'

但你必须小心：

>>> "png".split(".")[-1]
'png'    # But file doesn't have an extension

也不会在Unix系统中处理隐藏文件：

>>> ".bashrc".split(".")[-1]
'bashrc'    # But this is not an extension

对于一般用途，首选os.path.splitext

即使这个问题已经得到了回答，我也会在Regex中添加解决方案。

>>> import re
>>> file_suffix = ".*(\..*)"
>>> result = re.search(file_suffix, "somefile.ext")
>>> result.group(1)
'.ext'

您可以在pathlib模块（python3.x中提供）中找到一些很棒的东西。

import pathlib
x = pathlib.PurePosixPath("C:\\Path\\To\\File\\myfile.txt").suffix
print(x)

# Output 
'.txt'

filename='ext.tar.gz'
extension = filename[filename.rfind('.'):]