如何通过脚本让Google Colab实现单元格的条件跳转？

云上老周

2026-7-30

在Google Colab中实现单元格条件跳转与标签设置

一、条件跳转到下一个单元格

Google Colab没有原生Python API直接控制单元格跳转，但可以通过IPython结合JavaScript实现。以下是适配你需求的代码：

from IPython.display import display, Javascript

print("1 or 2")
choice = input("")

if choice == "1":
    # 触发跳转并执行下一个单元格
    display(Javascript('''
        const currentCell = google.colab.output.element.parentElement.parentElement;
        const nextCell = currentCell.nextElementSibling;
        if (nextCell) {
            google.colab.kernel.executeCell(nextCell.getAttribute('data-cell-index'));
        }
    '''))
    # 可选：停止当前单元格后续代码运行
    exit()
elif choice == "2":
    print("2")

这段代码的逻辑：

通过JavaScript获取当前单元格的DOM元素，找到其下一个相邻单元格
调用google.colab.kernel.executeCell执行目标单元格
加上exit()可以避免当前单元格在跳转后继续执行后续代码

二、为单元格添加标签并实现便捷跳转

你可以给单元格添加自定义标签，再通过代码跳转到指定标签的单元格：

1. 给单元格添加标签

右键点击目标单元格左侧的空白区域，或点击单元格右上角的「⋮」按钮，选择「Add tag」，输入自定义标签（例如data_processing_cell）。

2. 跳转至指定标签的单元格

使用以下代码实现跳转：

from IPython.display import display, Javascript

target_tag = "data_processing_cell"  # 替换为你的标签名
display(Javascript(f'''
    const targetCells = document.querySelectorAll('[data-tag~="{target_tag}"]');
    if (targetCells.length > 0) {{
        const cellIndex = targetCells[0].getAttribute('data-cell-index');
        google.colab.kernel.executeCell(cellIndex);
        // 平滑滚动到目标单元格
        targetCells[0].scrollIntoView({{behavior: 'smooth'}});
    }}
'''))