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

Question

我需要返回扩展名，所以字符串末尾有3个字母。当我运行这个时，什么也没有返回。当我注释掉第1行，并将return改为print时，它就工作了。如何使它与第1行活动和返回语句一起工作

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)]