获取文件的后缀名有好几种方法:
第一种:splittext()方法
os.path.splittext(path)[-1]
第二种:endswith()方法
path = "test_user_info.py"
bool = path.endswith(".py")
print(bool)
第三种:判断后缀名是否在字符串中(这种会存在误判,若是.pyx后缀,一样会打印True,前面两种不会)
path = "test_user_info.py"
if ".py" in path:
print(True)
第四种:用split方法切割,但是吧这种只是拿到了py没有点,所以再加上点也是可以的
path = "test_user_info.py"
suffix = path.split(".")[1]
print("suffix: {}".format(suffix))