如何在 python 方法中使用 globals() 更改全局变量？

Question

如果我有以下 Python 脚本 foo.py ：

def foo() :
  my_dict = {'A': 1, 'B': 2, 'C': 3}
  globals().update(my_dict)
  print(A)
  print(B)
  print(C)

foo()

我可以在 ipython 中输入这个：

run foo.py
=> 1
=> 2
=> 3
A
=> 1

但是，如果我删除底部的 foo() ：

def foo() :
  my_dict = {'A': 1, 'B': 2, 'C': 3}
  globals().update(my_dict)
  print(A)
  print(B)
  print(C)

# no foo

然后我在 ipython 中输入：

run foo.py
foo()
=> 1
=> 2
=> 3
A
=> NameError: name 'A' is not defined

我收到此错误。为什么会这样？

Answer 1

IPython 的%run 默认在新命名空间中运行文件，然后将结果复制到您的命名空间中。

当foo.py 包含foo() 调用时，会在新命名空间中创建A、B 和C 变量，然后IPython 会将这些变量复制到您的命名空间。

当foo.py 不包含foo() 调用时，您可以自己调用foo()，但foo 仍将其原始命名空间用于全局变量。调用 foo() 会在 foo 的原始命名空间中创建 A、B 和 C 变量，对于 IPython 来说太迟了，无法将它们拾取并将它们复制到您的命名空间中。

您可以使用%run 和-i 标志直接在您的命名空间中运行文件，避免这些问题。

Answer 2

您需要在分配之前声明element_dict 一个全局变量。一旦你在一个函数中有一个赋值语句anywhere，除非另有说明，否则被赋值的变量只在局部范围内声明。在您的情况下，由于element_dict = read_mass_from_file()，element_dict 在本地声明。

See the documentation for more details.

要实际引用全局变量：

def mass_mode() :
  '''
  Add element names and masses to the global variables.
  '''
  global element_dict
  element_dict = read_mass_from_file()

将从全局范围更新element_dict。