在Python中從文件名提取擴展名

是否有從文件名中提取擴展名的功能？

---

#1樓

一種選擇多是與點分開：

>>> filename = "example.jpeg"
>>> filename.split(".")[-1]
'jpeg'

文件沒有擴展名時沒有錯誤：

>>> "filename".split(".")[-1]
'filename'

但您必須當心：

>>> "png".split(".")[-1]
'png'    # But file doesn't have an extension

---

#2樓

值得在其中添加一個下標，這樣您就不會懷疑本身的列表爲何未顯示JPG。

os.path.splitext(filename)[1][1:].strip().lower()

---

#3樓

使用splitext時，雙擴展名的文件存在問題（例如file.tar.gz ， file.tar.bz2等）。

>>> fileName, fileExtension = os.path.splitext('/path/to/somefile.tar.gz')
>>> fileExtension 
'.gz'

但應爲： .tar.gz

可能的解決方案在這裏

---

#4樓

右拆分的另外一種解決方案：

# to get extension only

s = 'test.ext'

if '.' in s: ext = s.rsplit('.', 1)[1]

# or, to get file name and extension

def split_filepath(s):
    """
    get filename and extension from filepath 
    filepath -> (filename, extension)
    """
    if not '.' in s: return (s, '')
    r = s.rsplit('.', 1)
    return (r[0], r[1])

---

#5樓

filename='ext.tar.gz'
extension = filename[filename.rfind('.'):]