还有另一种更好的方法从python中的给定路径获取文件ext

Question

我是python中的新手，正如我们所知，我们可以通过使用从路径获取文件ext：

os.path.splitext(path)[1][1:]

这很好用，但似乎不太漂亮，我想问：是否有另一种更好的方法来获取文件分机

Answer 1

可读的方式是

name, ext = os.path.splitext(path)
ext = ext[1:]

虽然那不再是单一的表达。如果需要单个表达式，请将其包装在函数中：

def extension(path):
    name, ext = os.path.splitext(path)
    return ext[1:]

请注意，在某些系统上，文件可能具有空的扩展名，例如

>>> os.path.splitext('ham.')
('ham', '.')

并且您将那些与完全没有扩展名的文件相同（仅ham）。通常，这种差异并不重要，但在某些情况下可能会有所不同，这就是为什么splitext按照它的方式工作的原因。