python find file extension code example

Example 1: how to search for a specific file extension with python

import glob, os
os.chdir("/mydir")
for file in glob.glob("*.txt"):
    print(file)

Example 2: 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 3: 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 4: python get file extension from path

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

Example 5: os extension

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