如何优化提取文件扩展名并生成指定格式路径的Python代码？

你的代码逻辑没问题，但可以利用Python标准库中的工具实现更简洁、健壮的效果，同时能更好处理边缘场景（比如文件名无扩展名、包含多个点的情况），以下两种方案供参考：

方案一：使用os.path.splitext（传统路径处理） #

os.path.splitext是专门用于拆分文件名和扩展名的工具，会返回一个包含文件名主体和扩展名的元组，扩展名默认带.，我们只需去除前缀即可：

import os

file_name = str.casefold(input("Filename: ")).strip()
# 拆分文件名和扩展名，例如"test.png"会拆成("test", ".png")
root, ext = os.path.splitext(file_name)
# 去除扩展名的点，若没有扩展名则返回空字符串
ext = ext.lstrip('.')
# 拼接结果，无扩展名时可按需设置默认值
new_fname = f"image/{ext}" if ext else "image/unknown"
print(new_fname)

方案二：使用pathlib（面向对象路径处理，Python3.4+推荐） #

pathlib是Python3.4引入的现代路径处理库，语法更直观，支持面向对象操作：

from pathlib import Path

file_name = str.casefold(input("Filename: ")).strip()
file_path = Path(file_name)
# suffix属性获取带点的扩展名，lstrip去除点
ext = file_path.suffix.lstrip('.')
# 处理无扩展名的情况
new_fname = f"image/{ext}" if ext else "image/unknown"
print(new_fname)

对比原代码的优势 #

• 更贴合Python的惯用写法，代码可读性更强
• 自动处理文件名中多个点的情况（比如"archive.tar.gz"会提取gz作为扩展名，和你的原代码逻辑一致）
• 可以轻松处理无扩展名的场景，避免原代码中把整个文件名当作扩展名的问题

内容的提问来源于stack exchange，提问作者SoulSeeker916