python获取文件扩展名的方法

在使用Python进行文件操作的时候，我们经常需要获取文件的扩展名。例如，我们可能需要根据文件类型进行不同的处理，或者需要将文件重命名并保留原始的扩展名。本文将介绍多种Python获取文件扩展名的方法。

一、使用os.path.splitext方法

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

```python

import os

filename = 'example.txt'

ext = os.path.splitext(filename)[1]

print(ext)

```

输出结果为：.txt

二、使用os.path.basename方法

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

```python

import os

filename = 'example.txt'

basename = os.path.basename(filename)

ext = basename.split('.')[-1]

print(ext)

```

输出结果为：txt

三、使用splitext方法

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

```python

import ntpath

filename = 'example.txt'

ext = ntpath.splitext(filename)[1]

print(ext)

```

输出结果为：.txt

四、使用正则表达式

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

```python

import re

filename = 'example.txt'

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

print(ext)

```

输出结果为：txt

五、使用pathlib模块

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

```python

import pathlib

filename = 'example.txt'

p = pathlib.Path(filename)

ext = p.suffix

print(ext)

```

输出结果为：.txt

六、总结

本文介绍了多种Python获取文件扩展名的方法，包括使用os.path.splitext方法、os.path.basename方法、ntpath中的splitext方法、正则表达式和pathlib模块。这些方法都可以很方便地获取文件扩展名，可以根据具体需求选择使用。在实际应用中，我们需要根据不同的情况选择最合适的方法。