python中获取文件后缀名的方法

第一种:os.path.splitext()

# -*- coding: utf-8 -*- 
# @Author : yunze
import os
d = 0
path = r"E:\python project\generation\test"
srcfile = os.listdir(path)
for i in srcfile:
    print(os.path.splitext(i)[1])

第二种：endswith()方法

# -*- coding: utf-8 -*- 
# @Author : yunze
import os
d = 0
path = r"E:\python project\generation\test"
srcfile = os.listdir(path)
for i in srcfile:
    print(i.endswith(".txt"))

第三种：判断后缀名是否在字符串中（这种会存在误判，若是.pyx后缀，一样会打印True，前面两种不会）

path = r"E:\python project\generation\test"
if ".py" in path:
    print(True)

第四种：用split方法切割，但是吧这种只是拿到了py没有点，所以再加上点也是可以的

path = r"E:\python project\generation\test"
suffix = path.split(".")[1]
print("suffix: {}".format(suffix))