Python之在函數中使用列表作為默認參數


  本文將介紹筆者在工作中遇到的Python的一個坑,那就是使用列表作為默認參數。
  我們知道,在Python中,列表(list)是可變對象,所以列表的內容可能會在函數內改變。另一個需要注意的是,使用列表作為函數的默認參數時,列表的內容變化情況。
  首先,我們看以下的代碼例子:

def add(x, lst=[]):
    if x not in lst:
        lst.append(x)

    return lst

def main():
    list1 = add(1)
    print(list1)

    list2 = add(2)
    print(list2)

    list3 = add(3, [11, 12, 13, 14])
    print(list3)

    list4 = add(4)
    print(list4)

main()

你可能以為輸出的結果會是:

[1]
[2]
[11, 12, 13, 14, 3]
[4]

但事實上,該程序輸出的結果是:

[1]
[1, 2]
[11, 12, 13, 14, 3]
[1, 2, 4]

這是為什么呢?函數add的功能是當x不在列表中時,將x追加給列表lst。當函數第一次執行時,參數lst的默認值[]被創建。這個默認值只會被創建一次。add(1)將1加到lst。當函數再次被調用時,lst是[1]而不是[],因為lst只被創建一次。當參數的lst為[11,12,13,14]時,lst就是[11,12,13,14]。list4調用函數時,使用默認參數,因此,現在默認參數lst為[1,2]。
  為了更好地理解調用情況,可以在add函數中輸出lst的id,如以下代碼:

def add(x, lst=[]):
    print(id(lst))

    if x not in lst:
        lst.append(x)
    
    return lst

def main():
    list1 = add(1)
    print(list1)

    list2 = add(2)
    print(list2)

    list3 = add(3, [11, 12, 13, 14])
    print(list3)

    list4 = add(4)
    print(list4)

main()

輸出結果如下:

4469603648
[1]
4469603648
[1, 2]
4469670472
[11, 12, 13, 14, 3]
4469603648
[1, 2, 4]

可以看到,list1, list2, list4調用時默認參數的id並沒有發生改變,而list3的id發生了改變。
  這是Python使用列表作為默認參數的一個坑。那么,如何避免踩坑呢?如果想要使用默認列表在每次函數調用時都是[],可以像下面的程序那樣修改函數參數:

def add(x, lst=None):

    if lst is None:
        lst = []
    if x not in lst:
        lst.append(x)

    return lst

def main():
    list1 = add(1)
    print(list1)

    list2 = add(2)
    print(list2)

    list3 = add(3, [11, 12, 13, 14])
    print(list3)

    list4 = add(4)
    print(list4)

main()

輸出的結果如下:

[1]
[2]
[11, 12, 13, 14, 3]
[4]

本次分享到此結束,感謝閱讀~

注意:本人現已開通微信公眾號: Python爬蟲與算法(微信號為:easy_web_scrape), 歡迎大家關注哦~~


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM