如何查找文件名的特定部分/扩展名？

Question

我想查找文件名的扩展名,扩展名只能是3个字符.

它是这样的,

filename = str(input("Please enter filename: "))

然后我想执行一个任务来寻找扩展,如果扩展不满足要求(即3个字符长),我将添加打印"无效的扩展名!" 否则,请打印"有效扩展名".

Answer 1

len(os.path.splitext('/foo/file.png')[1]) - 1 == 3