Python自定义myExec函数中列表推导式变量未定义问题求助

问题原因 #

1. 列表推导式的作用域特性：Python 3中，列表推导式会创建独立的嵌套作用域，而非直接复用当前执行环境的命名空间。这种设计避免了变量泄漏，但也导致它的变量查找规则和普通表达式（比如print(a)或[a,a,a]）不同。
2. 原生exec的作用域处理：原生exec在执行代码时，会自动关联调用者的全局（globals）和局部（locals）命名空间，列表推导式的嵌套作用域能通过作用域链正确找到外层的变量。
3. 自定义myExec的缺陷：如果你的自定义myExec没有显式传递或维护正确的全局/局部命名空间（比如仅简单调用exec(code)而不指定globals和locals参数），列表推导式的嵌套作用域就无法找到外层的a变量，从而抛出NameError。

举个典型的错误实现：

def myExec(code):
    # 未指定globals和locals，exec会使用默认的空字典作为locals
    exec(code)

调用时，print(a)能正常运行是因为它直接查找全局命名空间的a，但列表推导式的嵌套作用域无法访问到这个全局变量。

解决方法 #

显式传递调用者的全局和局部命名空间给exec，让列表推导式的嵌套作用域能通过作用域链找到变量。

方法1：让自定义函数自动获取调用者的命名空间 #

利用inspect模块获取调用者的全局和局部命名空间：

import inspect

def myExec(code):
    # 获取调用者的全局和局部命名空间
    caller_frame = inspect.currentframe().f_back
    exec(code, caller_frame.f_globals, caller_frame.f_locals)

方法2：允许调用者显式传递命名空间 #

和原生exec保持一致，支持接收globals和locals参数：

def myExec(code, globals=None, locals=None):
    exec(code, globals, locals)

调用时直接传递当前的命名空间：

a = 10
myExec("u = [a for x in range(3)]", globals(), locals())

为什么[a,a,a]能正常运行？ #

这种写法是普通的列表字面量，不会创建嵌套作用域，代码直接在exec的当前作用域中查找a，因此即使没有显式传递命名空间，也能找到全局的a变量。

内容的提问来源于stack exchange，提问作者Roland