如何在Python中获取文件扩展名？

在Python中，我们可以使用下面讨论的两种不同方法中的任何一种来提取文件扩展名 -

方法一：使用Python os模块的splitext()函数

此函数将文件路径字符串拆分为文件名和文件扩展名，将其拆分为一对根和扩展名，这样当两者都添加时，我们可以再次检索文件路径（文件名 + 扩展名 = 路径）。当 OS 模块已被使用时，最好使用此函数。

例子：

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)

输出：

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

方法二：使用Pathlib模块

Pathlib 模块的 pathlib.Path().suffix 方法可用于提取文件路径的扩展名。这种方法更适合面向对象的方法。

例子：

蟒蛇3

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

输出：

File Extension:  .txt