获取文件扩展名 Python

在Python中，我们可以通过os.path模块中的函数os.path.splitext来获取文件的扩展名。

代码示例

import os

filename = 'example.txt'
extension = os.path.splitext(filename)[1]

print(extension)  # 输出: '.txt'

解析

上述代码中，我们首先导入了os模块，然后定义了一个文件名filename，并指定扩展名为.txt。接着，我们使用os.path.splitext函数来获取文件的扩展名extension。该函数会将文件名和扩展名分开，并以元组(tuple)的形式返回。我们需要用下标1来获取扩展名。

最后，我们打印出扩展名extension，并输出结果为.txt。

总结

通过上述示例，我们了解了如何使用Python获取文件的扩展名。使用os.path.splitext函数可以获取文件名和扩展名，从而简化了文件操作的流程。