Python - 从路径获取文件扩展名

当我们处理文件时，有时需要获取文件的扩展名，Python提供了多种方法可以实现获取文件扩展名的功能。在本文中，我们将学习从路径获取文件扩展名的几种方法。

方法一：使用os模块

通过使用os模块中的os.path.splitext()函数，我们可以很容易地从文件路径中获取文件扩展名。该函数将文件路径拆分为文件名和扩展名，并返回一个元组。

import os

file_path = "/home/user/documents/sample.txt"
file_ext = os.path.splitext(file_path)[1]

print(file_ext)   # 输出：.txt

方法二：使用split()函数

Python的字符串类具有许多有用的方法，其中包括split()函数。我们可以使用split()函数将文件路径拆分为文件名和扩展名。另外我们还可以使用os.path.basename()方法来获取文件名。

import os

file_path = "/home/user/documents/sample.txt"
file_name = os.path.basename(file_path)
file_ext = file_name.split(".")[-1]

print(file_ext)   # 输出：txt

方法三：使用pathlib模块

Python 3.4及以上版本引入了一个新的模块pathlib，它提供了一种面向对象的方法来处理文件路径。我们可以直接使用pathlib.Path()类来获取文件扩展名。

import pathlib

file_path = "/home/user/documents/sample.txt"
file_ext = pathlib.Path(file_path).suffix

print(file_ext)   # 输出：.txt

以上就是从路径获取文件扩展名的3种方法。

总结

无论您是学习Python的初学者还是经验丰富的开发人员，获取文件扩展名都是日常工作中经常遇到的任务。在本文中，我们研究了Python中从路径中获取文件扩展名的三种方法。 通过使用这些技术，您可以更加高效地处理文件并获得所需的信息。