定义函数问题并返回子字符串

Question

我需要返回扩展名，因此字符串的末尾是3个字母。当我运行此程序时，什么也不会返回。当我注释掉第1行并更改return to print时，它起作用了。如何在第1行处于活动状态和带有return语句时使其工作？

def parseExtension(filename):
    filename = "assignment5.pdf"
    return (filename[-3:])

Answer 1

只需使用

def parseExtension(filename):
    filename = "assignment5.pdf"
    filelist = filename.split('.')
    return filelist[len(filelist) - 1)]