獲取文件的后綴名有好幾種方法:
第一種: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))