如何在JupyterLab中通过编程修改单元格内容以锁定数据集ID

嘿，我刚好碰到过和你一模一样的需求！每次跑单元格拿最新数据做分析，过半小时再跑新的，结果回头重跑旧单元格就拿到新数据了，太闹心了。不过我找到了一个用IPython实现的完美解决办法，不用依赖外部工具，直接在单元格里写代码就能搞定。

先给你理清楚思路：我们要做的就是在单元格第一次运行时，把curr_id = get_curr_id()这行代码，自动替换成get_curr_id()返回的实际ID值，这样后续不管什么时候重跑这个单元格，都会用第一次锁定的那个ID，不会再去数据库拉最新的了。

直接上具体的代码实现，你可以把这段代码放到你的目标单元格里：

from IPython import get_ipython

def lock_dataset_id():
    # 获取当前的IPython shell实例，这是和Jupyter交互的核心对象
    shell = get_ipython()
    # 拿到当前正在执行的单元格的原始内容
    current_cell_content = shell.user_ns['_ih'][-1]
    # 执行函数拿到实际的数据集ID
    curr_id = get_curr_id()
    # 替换掉单元格里的动态获取ID的代码行
    updated_cell_content = current_cell_content.replace(
        "curr_id = get_curr_id()", 
        f"curr_id = {curr_id}"
    )
    # 把修改后的内容替换回当前单元格，界面上会直接显示更新后的代码
    shell.set_next_input(updated_cell_content, replace=True)

# 先正常执行获取数据的逻辑
curr_id = get_curr_id()
data = load_data(curr_id)
# 调用锁定函数，自动修改单元格内容
lock_dataset_id()

给你拆解一下这段代码的作用：

• 当你第一次运行这个单元格时，会先正常调用get_curr_id()拿到当前最新的ID，加载对应的数据。
• 然后lock_dataset_id()函数会自动把单元格里的curr_id = get_curr_id()替换成类似curr_id = 12345（这里的12345就是实际拿到的ID值）。
• 替换完成后，你会看到JupyterLab里的这个单元格内容已经变成了固定ID的版本，之后不管什么时候重跑它，都会用这个固定的ID加载原始的数据集，不会再去拉最新的了。

完全适配你的工作流：
你现在可以先跑这个单元格分析第一批数据，它会自动锁定ID；半小时后直接复制这个单元格，把里面的curr_id = 12345改回curr_id = get_curr_id()，就能拿到新的数据集做分析。回头再看第一个单元格，重跑它还是用原来的12345 ID，完美解决你担心的问题！

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