从Python文件名中提取扩展名

Question

从Python文件名中提取扩展名是否有从文件名中提取扩展名的函数？

Answer 1

是。使用os.path.splitext(见Python2.x文档或Python3.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没有扩展而没有扩展.c/d，它会治疗.bashrc没有扩展而没有扩展.bashrc:

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

Answer 2

import os.path
extension = os.path.splitext(filename)[1]