Python file extension code example

Example 1: split filename and extension python

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

Example 2: python check file extension

# Checking for single extension (.txt) of file
if file.endswith(".txt"):
  # do something
  
# Checking multiple extensions (.txt, .pdf, .mp3, ...)
if file.endswith((".txt", ".pdf", ".mp3")):
  # do something

Example 3: python get file extension from path

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

Example 4: Python file extension

filename = input("Input the Filename: ")
f_extns = filename.split(".")
print ("The extension of the file is : " + repr(f_extns[-1]))

Example 5: os extension

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