1. 程式人生 > >python函數中把列表(list)當參數時的"入坑"與"出坑"

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

[ ] 優化 list not 判斷 class tro 原來 PE

在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---
[0, 1, 4, 9]
---2---
[0, 1, 4, 9, 16]
---3---
[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]

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