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

Question

我是 python 的新手，正如我们所知，我们可以通过使用从路径获取文件扩展名:

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

这个很好用，但是好像不太漂亮，想问一下:有没有其他更好的方法获取文件ext

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 以这种方式工作的原因。