在 Python 中获取文件后缀(扩展名)有以下几种常用方法:
方法1:使用 os.path.splitext()
import os
filename = "example.txt"
extension = os.path.splitext(filename)[1]
print(extension) # 输出: .txt
方法2:使用 pathlib.Path(Python 3.4+推荐)
from pathlib import Path
filename = "example.txt"
extension = Path(filename).suffix
print(extension) # 输出: .txt
方法3:字符串分割(简单情况)
filename = "example.txt"
extension = filename.split(".")[-1] if "." in filename else ""
print(extension) # 输出: txt
# 如果需要带点号
extension_with_dot = "." + extension if extension else ""
注意事项:
- 这些方法会保留点号(如
.txt) - 对于没有扩展名的文件会返回空字符串
- 对于多个点号的文件(如
archive.tar.gz),os.path.splitext()只认最后一个扩展名 - 如果需要获取所有扩展名,可以使用
pathlib.Path的.suffixes属性:
from pathlib import Path
filename = "archive.tar.gz"
print(Path(filename).suffixes) # 输出: ['.tar', '.gz']
推荐使用 pathlib 方法,它是现代 Python 处理文件路径的首选方式。
1183

被折叠的 条评论
为什么被折叠?



