python如何识别后缀名

在Python中，你可以使用多种方法来识别文件的后缀名。以下是一些常见的方法：

1. 使用`os.path`模块

`os.path`模块中的`splitext`函数可以用来分割文件名和后缀名。

```python

import os

def get_extension(filename):

return os.path.splitext(filename)[1]

示例

filename = 'example.py'

extension = get_extension(filename)

print(extension) 输出: .py

```

2. 使用正则表达式

使用`re`模块可以编写一个简单的正则表达式来匹配文件名中的后缀名。

```python

import re

def get_extension(filename):

match = re.search(r'.([a-zA-Z0-9]+)$', filename)

if match:

return match.group(1)

return None

示例

filename = 'example.py'

extension = get_extension(filename)

print(extension) 输出: py

```

3. 使用`pathlib`

Python 3.4+引入了`pathlib`模块，它可以让你更方便地处理文件路径。

```python

from pathlib import Path

def get_extension(filename):

return Path(filename).suffix

示例

filename = 'example.py'

extension = get_extension(filename)

print(extension) 输出: .py

```

这些方法都可以有效地识别文件的后缀名。你可以根据需要选择最适合你的方法。