获取文件的后缀名有好几种方法:

第一种:splittext()方法

os.path.splitext(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))
Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐