在 python 函数中作为参数分配的变量名

Question

我正在编写一个使用treelib 库构建和填充树状数据结构的函数。您按如下方式创建树：

foo = Tree()

...然后从那里开始，这已经足够标准了。这是我的简化版：

def make_special_tree(tree, arg1, arg2):
    tree = Tree()
    other_stuff = things(arg1)
    modify_tree(tree, other_stuff, arg2)
    return tree

事情是这样的：假设我最终想要一个名为blah 的树对象。如果我执行以下操作，则命令运行时不会出现错误：

make_special_tree('blah', foo, bar)

...但是当我之后输入blah 时，我会返回NameError: name 'blah' is not defined。如果我执行以下操作，该命令也会运行而不会出现错误：

blah = make_special_tree('yoink', foo, bar)

...然后当我输入blah 时，我会返回<treelib.tree.Tree object at 0x10e60db10>，这就是我想看到的。 yoink 同时保持未定义，就像之前版本中的 blah。

因此我的问题是——我可以说这是基本的，但我无法解开这个问题，部分原因是我不确定问这个问题的准确程度。如您所见，现在我必须创建类 Tree() 的实例，并且我认为我必须为我的函数提供一个参数才能这样做。我认为blah = make_special_tree(args) 是格式化它的正确方法，但我怎样才能将变量blah 作为我想要返回的树结构的名称传递？

Answer 1

Python 函数通常不会“分配变量名”，它们只是返回值，您可以对这些值做任何您想做的事情。

一个值没有变量名。^* 因为一个值可以存储在五个不同的变量中，或者它可以存储在非变量的东西中，比如列表。没关系。

所以，你几乎肯定想做的是@spectras 在评论中建议的：去掉第一个参数，然后将值存储在你想要的任何变量中：

def make_special_tree(arg1, arg2):
    tree = Tree()
    other_stuff = things(arg1)
    modify_tree(tree, other_stuff, arg2)
    return tree

# Two different trees with different names
blah = make_special_tree(foo, bar)
yoink = make_special_tree(foo, bar)

# One tree with two names
spam = make_special_tree(foo, bar)
eggs = spam

# A tree with no names at all
cheese = []
cheese.append(make_special_tree(foo, bar))

---

如果你真的想创建一个具有指定名称的全局变量，你可以这样做，^** 但这是一个非常糟糕的主意，除非你试图做一些不寻常的事情，比如collections.namedtuple 的方式类似于 class 语句。

---

_{* 一个值可以有一个名字，例如，一个name 属性。您甚至可以更改Tree 类的定义，以将该name 属性用作其输出表示的一部分。但这仍然与你存储它的变量无关。}

_{** 最简单的方式是globals()[name] = value。}

Answer 2

函数中的第一个参数是多余的。试试这个：

def make_special_tree(arg1, arg2):
    tree = Tree()
    other_stuff = things(arg1)
    modify_tree(tree, other_stuff, arg2)
    return tree

blah = make_special_tree(foo, bar)