Python：設計函數實現-返回給定文件名的後綴

rfind()方法：返回字符串最後一次出現的位置（下標），如果沒有匹配項就返回-1

 

方法內容如下：

 

def get_suffix(filename):
    """
    :param filename: 文件名稱
    :return: 返回的後綴名
    """
    pos = filename.rfind('.')
    print('.出現的下標是%d' % pos)
    # 如果沒有匹配項返回-1，不等於-1就有後綴名
    if pos != -1:
        print(filename[pos:])
    else:
        print('輸入錯誤')

驗證是否可行：

 

if __name__ == '__main__':
    file_name = input('請輸入文件名')
    get_suffix(file_name)

 

效果如下：