Skip to content
Related Articles

Related Articles

Improve Article
How to get file extension in Python?
  • Last Updated : 16 Nov, 2020

In Python, we can extract the file extension using either of the two different approaches discussed below –

Method 1: Using Python os module splitext() function

This function splits the file path string into file name and file extension into a pair of root and extension such that when both are added then we can retrieve the file path again (file_name + extension = path). This function is preferred to use when the OS module is being used already.

Example:

Python3






import os
  
  
# this will return a tuple of root and extension
split_tup = os.path.splitext('my_file.txt')
print(split_tup)
  
# extract the file name and extension
file_name = split_tup[0]
file_extension = split_tup[1]
  
print("File Name: ", file_name)
print("File Extension: ", file_extension)

Output:

('my_file', '.txt')
File Name:  my_file
File Extension:  .txt

Method 2: Using Pathlib module

pathlib.Path().suffix method of the Pathlib module can be used to extract the extension of the file path. This method is preferred for an object-oriented approach.

Example:

Python3




import pathlib
  
# function to return the file extension
file_extension = pathlib.Path('my_file.txt').suffix
print("File Extension: ", file_extension)

Output:

File Extension:  .txt

 Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.  

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning – Basic Level Course




My Personal Notes arrow_drop_up
Recommended Articles
Page :