当前位置:首页 > 行业动态 > 正文

python如何识别后缀名

在Python中,我们可以使用os模块的splitext()函数来识别文件的后缀名,这个函数会将文件名和扩展名分开,返回一个元组,其中第一个元素是文件名,第二个元素是扩展名。

以下是一个简单的示例:

import os
def get_file_extension(filename):
    return os.path.splitext(filename)[1]
测试
print(get_file_extension('example.txt'))  # 输出: .txt
print(get_file_extension('example.tar.gz'))  # 输出: .gz

在这个示例中,我们定义了一个函数get_file_extension(),它接受一个参数filename(文件名),然后使用os.path.splitext()函数来获取文件的扩展名,我们打印出了两个测试用例的结果。

0

随机文章