获取文件扩展名是Python中常见的操作之一,可以使用以下两种方法来获取文件扩展名:
方法一:使用split()方法
使用split()方法可以将文件名拆分成文件名和扩展名,并返回一个列表,其中第二个元素即为扩展名。
file_name = "example.jpg"
file_ext = file_name.split(".")[-1]
print(file_ext) # 输出:jpg
上述代码中,首先将文件名"example.jpg"作为字符串存储在变量file_name中,然后使用split()方法,将文件名按照"."进行分割,返回一个列表,最后获取列表的最后一个元素,即为扩展名。
方法二:使用os模块中的os.path.splitext()方法
os模块中的os.path.splitext()方法可以直接将文件名与扩展名拆分,并返回一个二元组,第二个元素即为扩展名。
import os
file_name = "example.jpg"
file_ext = os.path.splitext(file_name)[-1]
print(file_ext) # 输出:.jpg
上述代码中,首先引入os模块,然后将文件名"example.jpg"作为字符串存储在变量file_name中,接着使用os.path.splitext()方法,将文件名与扩展名拆分成二元组,最后获取二元组的第二个元素,即为扩展名。
需要注意的是,使用os.path.splitext()方法获取的扩展名包含"."符号。
综上所述,Python中获取文件扩展名的方法有两种,分别是使用split()方法和os.path.splitext()方法。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:python获取文件扩展名的方法 - Python技术站