Jupyter Notebook跨文件调用：实现独立变量空间与指定变量的传递接收

我完全理解你遇到的困扰——%run -i会无差别共享所有变量，nbclient的方案又踩了坑，转换为py文件还要付出额外的迁移成本。下面这个方案可以精准满足你的所有需求：

• 两个笔记本拥有完全独立的变量空间，不会互相污染
• 手动指定要传递给sub的变量，也只提取你需要的变量返回
• 全程不需要将ipynb转换为py文件，直接在内存中处理sub的代码

核心思路 #

通过nbformat读取sub.ipynb的代码单元格，在一个独立的Python字典作为全局命名空间中执行代码：

1. 只将你指定的变量传入这个独立命名空间
2. 所有sub的代码都在这个空间内运行，完全不影响main的变量
3. 执行完成后，从这个命名空间中提取你需要的变量返回给main

完整示例代码 #

1. sub.ipynb（被调用的子笔记本）

这个笔记本负责接收传入的变量、执行业务逻辑，生成需要返回的变量：

# 接收main传入的变量x（仅来自指定的独立命名空间，与main的x完全无关）
print(f"sub笔记本接收到的x值: {x}")

# 示例业务逻辑（替换为你自己的复杂代码即可）
y = x * 2
z = x + 3  # 这个变量我们不需要返回给main，不会被提取

# 支持任何Jupyter代码：数据处理、可视化、甚至魔法命令（兼容方案见下文）

2. main.ipynb（主笔记本）

这段代码实现sub的调用、变量传递与结果提取：

import nbformat
from IPython import get_ipython

def run_sub_notebook(sub_nb_path, input_vars=None, output_vars=None):
    """
    运行指定Jupyter Notebook，精准控制变量的传入与传出
    :param sub_nb_path: 子笔记本路径（例如"./sub.ipynb"）
    :param input_vars: 要传入的变量字典，格式如{"x": 5}
    :param output_vars: 要提取的变量列表，格式如["y"]
    :return: 提取到的结果字典，格式如{"y": 10}
    """
    # 1. 读取子笔记本的全部内容
    nb = nbformat.read(sub_nb_path, as_version=4)
    
    # 2. 创建独立的命名空间，仅传入指定变量
    sub_namespace = {}
    if input_vars:
        sub_namespace.update(input_vars)
    
    # 3. 逐单元格执行sub的代码（支持魔法命令）
    ip = get_ipython()
    for cell in nb.cells:
        if cell.cell_type == 'code':
            try:
                # 在独立命名空间中运行代码，支持IPython魔法命令
                ip.run_cell(
                    cell.source,
                    store_history=False,
                    silent=False,
                    user_ns=sub_namespace
                )
            except Exception as e:
                print(f"执行sub笔记本代码出错: {str(e)}")
                raise
    
    # 4. 提取指定的返回变量
    result = {}
    if output_vars:
        for var_name in output_vars:
            if var_name in sub_namespace:
                result[var_name] = sub_namespace[var_name]
            else:
                print(f"警告：sub笔记本中不存在变量 {var_name}")
    
    return result

# ------------------- 测试调用 -------------------
# main中的独立变量，与sub完全隔离
x_in_main = 1
print(f"main初始x值: {x_in_main}")

# 调用sub：传入x=5，仅提取y变量
sub_result = run_sub_notebook(
    "./sub.ipynb",
    input_vars={"x": 5},
    output_vars=["y"]
)

# 查看返回结果
print(f"从sub获取的y值: {sub_result['y']}")

# 验证变量隔离：main的x值完全不受sub影响
print(f"调用后main的x值仍为: {x_in_main}")

方案优势与注意事项 #

1. 绝对变量隔离：sub的所有变量都在独立字典中运行，不会污染main的变量空间，反之亦然
2. 精准变量控制：你可以自由选择传入/传出的变量，完全避免变量混乱
3. 兼容复杂场景：支持sub中的所有Jupyter代码（包括魔法命令、可视化、数据处理等）
4. 零文件转换：无需将ipynb转为py文件，直接读取原笔记本内容执行

无魔法命令场景的简化版（可选） #

如果你的sub笔记本中没有IPython魔法命令，可以将代码执行部分替换为更轻量的exec，不需要依赖IPython API：

# 替换main中执行代码的循环部分
for cell in nb.cells:
    if cell.cell_type == 'code':
        try:
            exec(cell.source, sub_namespace)
        except Exception as e:
            print(f"执行sub笔记本代码出错: {str(e)}")
            raise

备注：内容来源于stack exchange，提问作者PTQuoc