我们如何修改传递给Python函数的对象

Question

我创建了一个小函数function，如果传递-1的值，则应将none分配给节点，然后将值分配给节点对象的值属性。我创建了一个简单的二进制树[根，左，右]，最后打印了节点。

class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

def funct(node, val):
    if val == -1:
        ### modify code here 
        node = None
    else:
        node.val = val

root = TreeNode()
root.val = 'root'
root.left = TreeNode()
root.right = TreeNode()

funct(root.left, 'left')
funct(root.left, -1)

print(root, root.val)
print(root.left, root.left.val)
print(root.right, root.right.val)

当我打印节点时，我会看到以下输出。

正确的节点在内存中，不是无。

如何通过修改中的代码（ in function> function中）以获取以下输出，从而将无分配给Orignal对象。

大量模拟以下代码并获取输出：

root = TreeNode()
root.val = 'root'
root.left = TreeNode('left')
root.right = None

注意：我可以更改我的算法。仅在Val！= -1时创建一个新节点。但是，我想了解是否有一种方法可以在Pyhton修改经过的对象。

编辑：删除“删除”一词。添加了一些澄清。

Answer 1

i创建了一个小函数，如果传递-1的值，该函数应分配给节点，否则将值分配给节点对象的值属性。

funct无法替换传递给它的节点（即：使呼叫者的变量之一是指新节点，而不是该节点），因为它接收 node ，和不是该节点的任何变量名称。

它可以修改传递给它的节点。特别是，它可以通过重新分配到node.right和node.left.left来替换子值。这意味着“ .right和.left的treenode实例传递给我”。 node是一个名称function用于该传递实例的名称，与 无关的是呼叫代码 。

那么，要“删除”子树，我们需要通过将要删除的树的 parent parent，并指示将删除哪一侧。看起来可能是：

def remove_child(parent, which):
    if which == 'left':
        parent.left = None
    elif which == 'right':
        parent.right = None
    else:
        raise ValueError("invalid child name")

树的根没有父母，因此除去整棵树必须被视为特殊情况。

Answer 2

不确定这是否是您需要的，但我认为它可以做您想要的行为：

class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

    def funct(self, child: str, val):
        if val == -1:
        ### modify code here 
            setattr(self,child, None)
    
        else:
            setattr(self,child,TreeNode(val))


root = TreeNode()
root.val = 'root'
root.left = TreeNode()
root.right = TreeNode()

root.funct(child='left', val='left')
root.funct(child='right', val='right')

print(root, root.val)
print(root.left, root.left.val)
print(root.right, root.right.val)
print(None)

root.funct(child='left', val= -1)
print("\n")
print(root, root.val)
# print(root.left, root.left.val) - you cannot print root.left.val because None has no attribute
print(root.left)
print(root.right, root.right.val)