Python的類成員變數預設初始值的坑及解決

2022-02-11 16:00:07

類成員變數預設初始值的坑

問題發現

一個迴圈內，預設值初始化同名變數，其中的list成員不是空，會延續之前同名變數的值。

範例程式碼

# Define class
class Variant():
    # use
    def __init__(self, price = 500, description = 'default description', values = ['', '', '']):
        self.price = price
        self.description = description
        self.values = values
    
    def __str__(self):
        return 'price: {}, description: {}, values: {}'.format(self.price, self.description, self.values)

variant_list = []
# Create instance with same name iteratively
for i in range(3):
    current_variant = Variant()
    if i == 1:
        current_variant.values[2] = 'hello'
    current_variant.price = i
    current_variant.description = 'description of variant: {}'.format(i)
    variant_list.append(current_variant)
    
# Test results
for variant in variant_list:
    print(str(variant))

結果

所有範例的values列表值相同

原因

可選引數預設值的設定在Python中只會被執行一次，也就是定義該函數的時候”如此使用預設值初始化，list成員指向的是同一個list（地址），如果只是修改其中一個元素（而不是賦值新的list開闢新記憶體），那麼所有instance的list成員都會被修改。

解決方法

直接在構造方法中置為空（self.values = ['', '', '']），之後各個修改值

Python預設值引數

簡單粗暴上程式碼

def fun(a, b=[]):
    b += [a]
    print(b)

fun(1)
fun(2,[])
fun(3)

是不是看上去很簡單，其實暗藏玄機，請大家看一下輸出結果，是不是有點讓你疑惑^^~

[1]
[2]
[1, 3]

此時你是否也和我有一樣的疑惑，為什麼 fun(3) 的輸出結果是 [1, 3]？

哈哈，不賣關子了，這裡是因為，因為函數被定義好後，只會生成一次，所以在函數生成的時候定義的變數 b 的預設值也只會被初始化一次。

因此，當執行fun(1)函數時，沒有給 b 傳參，所以使用的是 b 的預設值，此時 b 的預設值為[1]。

執行fun(2,[])時，給 b 傳了一個[]值(恰好和預設值相同，其實是不同的資料)，因此便使用的是傳入資料，執行結果便是[2]。

然後在執行fun(3)，此刻又沒有給 b 傳參，所以依舊使用的是 b 的預設值，而 b 的預設值只會隨著函數的生成被生成一次 ( fun(1) 生成過了 )，所以現在的預設值是fun(1)的執行結果[1]，因此當fun(3)再次呼叫時，輸出結果便會是[1, 3]。

如果不行出現當前這種情況，而是在函數每次被呼叫的時候都初始化一次變數

可以用下面這種寫法

def function(a, b=None):
    b = b if b else []  # 明確每次重新定義b
    b += [a]
    print(b)

function(1)
function(2, [])
function(3)

輸出結果：

[1]
[2]
[3]

以上為個人經驗，希望能給大家一個參考，也希望大家多多支援it145.com。