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

name_only=file_name[:filename.index(".")

这将为您提供文件名，最多为第一个“.”，这将是最常见的。

虽然这是一个古老的话题，但我想知道为什么在本例中没有提到一个非常简单的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])

如果你喜欢正则表达式，那就是一个真正的单行。即使中间有额外的“.”也无关紧要

import re

file_ext = re.search(r"\.([^.]+)$", filename).group(1)

查看此处查看结果：单击此处