python函數中把列表(list)當參數時的"入坑"與"出坑"

在Python函數中,傳遞的參數如果默認有一個為 列表(list),那么就要注意了,此處有坑.

入坑

挖坑

def f(x,li=[]):
    for i in range(x):
        li.append(i*i)
    print(li)
 
print('---1---')
f(4)
print('---2---')
f(5)

預期結果

---1---
[0, 1, 4, 9] ---2--- [0, 1, 4, 9, 16]

執行結果

---1---
[0, 1, 4, 9]
---2---
[0, 1, 4, 9, 0, 1, 4, 9, 16]

出坑

當定義函數時，會保存函數中默認參數 list 的值，也就是列表 li=[]；

在每次調用的時候如果傳遞了新的列表，則使用傳遞的列表，沒有傳遞，使用定義函數時保存的默認參數（li=[]）；

上面兩次調用中，都沒有傳遞新的列表（使用默認列表 li=[] ），程序會調用定義函數時保存的默認參數(（li=[]）);

列表在append的時候會在 li=[] 原來的基礎上append追加值，所以會產生以上結果.

通過打印列表的ID進行辨識

打印列表 li=[] 的ID:

def f(x,li=[]):
    print(id(li))  ＃ 添加打印id
    for i in range(x):
        li.append(i*i)
    print(li)
 
 
print('---1---')
f(4)
print('---2---')
f(5)

結果：

---1---
140306123906248
[0, 1, 4, 9]
---2---
140306123906248
[0, 1, 4, 9, 0, 1, 4, 9, 16]

會發現ID值是相同的;

說明兩次執行時使用的都是定義函數時的默認參數 li=[ ]

執行時往里面傳新的列表

打印列表 li=[] 的ID 和 傳的新列表的ID:

def f(x,li=[]):
    print(id(li))
    for i in range(x):
        li.append(i*i)
    print(li)
 
 
print('---1---')
f(4)
print('---2---')
f(5,[])
print('---3---')
f(6)

結果：

---1---
140017293614280
[0, 1, 4, 9]
---2---
140017293614472
[0, 1, 4, 9, 16]
---3---
140017293614280
[0, 1, 4, 9, 0, 1, 4, 9, 16, 25]

會發現執行傳遞空(新)列表的函數時打印的ID不一樣,而沒有傳遞的一樣;

當傳遞空列表時,函數體當中會使用傳遞的空列表，沒有傳遞時，使用函數默認值 li=[ ], 所以會產生以上結果.

優化

如果想要達到預期的結果,只需要在函數體里進行判斷即可:

def f(x, li=[]):
    if not li:
        # 如果li不為空的話,就往下走(清空列表); 為空就不走
        li = []
    for i in range(x):
        li.append(i * i)
    print(li)


print('---1---')
f(4)
print('---2---')
f(5)
print('---3---')
f(6)

結果：

---1---
[0, 1, 4, 9]
---2---
[0, 1, 4, 9, 16]
---3---
[0, 1, 4, 9, 16, 25]