遞歸函數的調用,求嵌套列表的全部長度之和

def nest_list_len(input_list):

    if isinstance(input_list,list):
        length = 0
        length = length +len(input_list)
        for new_list in input_list:
            if isinstance(new_list,list):
                length += nest_list_len(new_list)
        return length
    else:
        return 0

if __name__ == '__main__':
    # 返回a中全部list元素的長度的和。4 + 3 + 3 = 10
    a = ['spam!', 1, ['Brie', 'Roquefort', 'Pol le Veq'], [1, 2, 3]]
    print nest_list_len(a)  # 輸出10

    a = ['spam!', 1,[['Brie'], 'Roquefort', 'Pol le Veq'], [1, 2, 3]]
    print nest_list_len(a)  # 輸出11

    a = 5  # 輸出0
    print nest_list_len(a)  # 輸出0

 

isinstance(a,list):判斷a是不是一個列表;spa

len(a):求a的長度。blog

相關文章
相關標籤/搜索