在Python中从文件名提取扩展名

Question

是否有从文件名中提取扩展名的功能？

Answer 1

使用os.path.splitext（请参阅Python 2.X文档或Python 3.X文档）：

>>> import os
>>> filename, file_extension = os.path.splitext('/path/to/somefile.ext')
>>> filename
'/path/to/somefile'
>>> file_extension
'.ext'

与大多数手动字符串拆分尝试不同，os.path.splitext它将正确地/a/b.c/d视为没有扩展而不是具有extension .c/d，并且将被.bashrc视为没有扩展而不是具有extension .bashrc：

>>> os.path.splitext('/a/b.c/d')
('/a/b.c/d', '')
>>> os.path.splitext('.bashrc')
('.bashrc', '')