追加功能将元素添加到列表的成员

问题描述

如何在不修改成员的情况下将Class对象添加到列表中？

class Node:
    def __init__(self,name,type,children=[]):
        self.name = name
        self.type = type
        self.children = children

    def add_child(self,child_node):
        self.children.append(child_node)

    def printNode(self,indent = 0):
        print(self.name + " has children " + str(len(self.children)))
        #for child in self.children: print(child,indent + 4)
            
if __name__ == "__main__":
    A = Node("A","company")
    B = Node("B","department")
    B.printNode()
    A.add_child(B)
    B.printNode()

append()函数将节点B添加到自身中，即使它仅应将其添加到节点A的子级列表中（从输出中可以明显看出）

B has children 0
B has children 1

解决方法

切勿使用可变数据（例如列表）作为默认参数。像这样重新定义您的构造函数，它将按预期工作

def __init__(self,name,type,children=None):
    self.name = name
    self.type = type
    self.children = children if children else []

,

在Python中定义函数参数时，您偶然发现了一个非常常见的陷阱。 here有一个很好的解释。

定义函数时，Python解释器将创建用作默认参数的值。然后，将这些默认值用于每个函数调用。在这种情况下，解释children=[]时将创建空列表__init__。然后，无论调用哪个对象，每次调用add_child()时都使用相同的列表。

这是一个例子：

def my_func(a_list=[]):
    a_list.append("something")
    print(a_list)

>>> my_func()
["something"]
>>> my_func()
["something","something"]

在这种情况下，a_list在每个函数调用中都会重复使用，并且大小会继续增长。

解决此问题的方法是使用一个不同的值来表示Node应该以空子代列表开头。像这样：

class Node:
    def __init__(self,children=None):
        self.name = name
        self.type = type
        self.children = children or []

...