在 Python 中有几种获取文件扩展名的方法。这里有三个选项:
- 使用os.path模块:
import os
filename = 'example.txt'
extension = os.path.splitext(filename)[1]
print(extension) # Output: '.txt'
- 使用该pathlib模块(在 Python 3.4 及更高版本中可用):
from pathlib import Path
filename = 'example.txt'
extension = Path(filename).suffix
print(extension) # Output: '.txt'
- 使用字符串操作:
filename = 'example.txt'
extension = filename[filename.rindex('.'):]
print(extension) # Output: '.txt'
注意:在这种情况下,我们使用rindex()方法找到’.'字符最后一次出现的索引,然后我们使用字符串切片来获取从该索引到字符串末尾的字符。这种方法的效率不如前两种,但对于简单的情况可能已经足够了。