python中获取文件扩展名

在 Python 中有几种获取文件扩展名的方法。这里有三个选项：

1. 使用os.path模块：

import os

filename = 'example.txt'
extension = os.path.splitext(filename)[1]
print(extension)  # Output: '.txt'

2. 使用该pathlib模块（在 Python 3.4 及更高版本中可用）：

from pathlib import Path

filename = 'example.txt'
extension = Path(filename).suffix
print(extension)  # Output: '.txt'

3. 使用字符串操作：

filename = 'example.txt'
extension = filename[filename.rindex('.'):]
print(extension)  # Output: '.txt'

注意：在这种情况下，我们使用rindex()方法找到’.'字符最后一次出现的索引，然后我们使用字符串切片来获取从该索引到字符串末尾的字符。这种方法的效率不如前两种，但对于简单的情况可能已经足够了。