python 获取文件扩展名

一、使用os.path.splitext方法

os.path.splitext方法是Python中常用的获取文件扩展名的方法之一。该方法将文件路径分割成路径和文件扩展名两个部分，并返回它们的元组。下面是一个例子：

import os

filename = 'example.txt'
ext = os.path.splitext(filename)[1]
print(ext)

输出结果为：.txt

二、使用os.path.basename方法

os.path.basename方法可以返回文件名及其扩展名。然后我们可以使用字符串分割方法来获取扩展名。下面是一个例子：

import os

filename = 'example.txt'
basename = os.path.basename(filename)
ext = basename.split('.')[-1]
print(ext)

输出结果为：txt

三、使用splitext方法

Python的内置模块ntpath中提供了splitext方法，可以直接获取文件扩展名。下面是一个例子：

import ntpath

filename = 'example.txt'
ext = ntpath.splitext(filename)[1]
print(ext)

输出结果为：.txt

四、使用正则表达式

正则表达式是一种强大的文本匹配工具，可以用来获取文件扩展名。下面是一个例子：

import re

filename = 'example.txt'
ext = re.findall('\.([^.]+)$', filename)[0]
print(ext)

输出结果为：txt

五、使用pathlib模块

Python的pathlib模块提供了一种面向对象的方法来处理文件路径和名称。可以使用pathlib.Path对象的suffix属性来获取文件扩展名。下面是一个例子：

import pathlib

filename = 'example.txt'
p = pathlib.Path(filename)
ext = p.suffix
print(ext)

输出结果为：.txt