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

虽然这是一个古老的话题，但我想知道为什么在本例中没有提到一个非常简单的python api rpartition：

要获取给定文件绝对路径的扩展名，只需键入：

filepath.rpartition('.')[-1]

例子：

path = '/home/jersey/remote/data/test.csv'
print path.rpartition('.')[-1]

将给您：“csv”

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

另一种右拆分解决方案：

# to get extension only

s = 'test.ext'

if '.' in s: ext = s.rsplit('.', 1)[1]

# or, to get file name and extension

def split_filepath(s):
    """
    get filename and extension from filepath 
    filepath -> (filename, extension)
    """
    if not '.' in s: return (s, '')
    r = s.rsplit('.', 1)
    return (r[0], r[1])

试试看：

files = ['file.jpeg','file.tar.gz','file.png','file.foo.bar','file.etc']
pen_ext = ['foo', 'tar', 'bar', 'etc']

for file in files: #1
    if (file.split(".")[-2] in pen_ext): #2
        ext =  file.split(".")[-2]+"."+file.split(".")[-1]#3
    else:
        ext = file.split(".")[-1] #4
    print (ext) #5

获取列表中的所有文件名拆分文件名并检查倒数第二个扩展名，它是否在penext列表中？如果是，则使用最后一个扩展名连接它，并将其设置为文件的扩展名如果没有，则只将最后一个扩展名作为文件的扩展名然后检查一下

使用splitext时，具有双扩展名的文件存在问题（例如file.tar.gz、file.tar.bz2等）

>>> fileName, fileExtension = os.path.splitext('/path/to/somefile.tar.gz')
>>> fileExtension 
'.gz'

但应该是：.tar.gz

可能的解决方案如下