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

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

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

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

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

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

# Output 
'.txt'

为了好玩。。。只需收集dict中的扩展，并在文件夹中跟踪所有扩展。然后，只要拉动你想要的延伸部分。

import os

search = {}

for f in os.listdir(os.getcwd()):
    fn, fe = os.path.splitext(f)
    try:
        search[fe].append(f)
    except:
        search[fe]=[f,]

extensions = ('.png','.jpg')
for ex in extensions:
    found = search.get(ex,'')
    if found:
        print(found)

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