将清理命名空间的工具函数放入模块后失效的解决办法

你遇到的核心问题是：当函数放在独立模块中时，globals()获取的是模块自身的全局命名空间，而非调用该函数的Jupyter Notebook的命名空间。下面提供两种优雅的解决方式：

方案一：显式传递调用者的全局命名空间 #

这是最直接清晰的实现方式，让调用者主动传入自身的globals()对象，函数基于传入的命名空间进行变量清理：

# my_utils.py
def clean_namespace(variables_to_delete, globals_dict):
    deleted = []
    for variable in variables_to_delete:
        if variable in globals_dict:
            del globals_dict[variable]
            deleted.append(variable)
    return deleted

在Jupyter Notebook中调用时：

from my_utils import clean_namespace
# 传入当前笔记本的全局命名空间
clean_namespace(['var1', 'var2'], globals())

方案二：自动获取调用者的全局命名空间 #

借助inspect模块获取调用栈的上下文，自动拿到调用者的全局命名空间，这样调用时无需额外传参，更便捷：

# my_utils.py
import inspect

def clean_namespace(variables_to_delete):
    # 获取调用者的栈帧（index=1对应调用当前函数的上层代码）
    caller_frame = inspect.currentframe().f_back
    caller_globals = caller_frame.f_globals
    
    deleted = []
    for variable in variables_to_delete:
        if variable in caller_globals:
            del caller_globals[variable]
            deleted.append(variable)
    
    # 手动删除帧引用，避免潜在内存泄漏
    del caller_frame
    return deleted

在Jupyter Notebook中直接调用：

from my_utils import clean_namespace
clean_namespace(['var1', 'var2'])

两种方案对比 #

• 方案一：无额外依赖，逻辑透明，适合需要明确控制命名空间的场景
• 方案二：调用更简洁，无需手动传参，但依赖Python标准库的inspect模块，需要注意清理栈帧引用避免内存问题

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