从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'

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

>>> 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'

上面的任何解决方案都有效，但在linux上，我发现扩展字符串末尾有一个换行符，这将阻止匹配成功。将strip（）方法添加到末尾。例如：

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

只需加入所有pathlib后缀。

>>> x = 'file/path/archive.tar.gz'
>>> y = 'file/path/text.txt'
>>> ''.join(pathlib.Path(x).suffixes)
'.tar.gz'
>>> ''.join(pathlib.Path(y).suffixes)
'.txt'

使用os.path.splitext：

>>> import os
>>> filename, file_extension = os.path.splitext('/path/to/somefile.ext')
>>> filename
'/path/to/somefile'
>>> file_extension
'.ext'

与大多数手动字符串拆分尝试不同，os.path.splitext将正确地将/a/b.c/d视为没有扩展名而不是扩展名.c/d，并将.bashrc视为没有延伸名而不是具有扩展名.bashrc：

>>> os.path.splitext('/a/b.c/d')
('/a/b.c/d', '')
>>> os.path.splitext('.bashrc')
('.bashrc', '')