Python輸出文件擴展名

def file_ext(filename):
    ls=[]
    for i in range(len(filename)):
        if (filename[i] in '.'):
            for a in range(i+1,len(filename)):
                ls.append(filename[a])
    return(''.join(ls))
filename = str(input())
print(file_ext(filename))