如何在IPython的cell_magic中获取单元格ID

嘿，这个问题我做IPython扩展时也碰到过！确实默认的@cell_magic只会把行参数和单元格内容传给你，要拿到单元格的唯一标识得绕个小弯，但IPython内核内部其实已经存了这些元数据。

核心思路是利用魔法类里的self.shell对象——它是IPython内核的实例，里面的parent_header属性包含了来自前端（比如Jupyter Notebook/Lab）的请求元数据，其中就有我们需要的cell_id。这个ID是每个单元格的唯一标识，单元格创建后哪怕反复编辑内容，它都不会改变，刚好能满足你区分“新单元格”和“编辑后重新执行”的需求。

给你个直接能用的代码示例：

from IPython.core.magic import magics_class, cell_magic

@magics_class
class CellTrackerMagics:
    @cell_magic
    def track_cell(self, line, cell):
        # 从内核请求元数据中提取cell_id
        cell_id = self.shell.parent_header.get('cell_id')
        
        if not cell_id:
            print("警告：无法获取单元格ID，可能不在Jupyter Notebook/Lab环境中")
            return
        
        # 维护一个集合记录已处理过的单元格ID
        if not hasattr(self, 'processed_cell_ids'):
            self.processed_cell_ids = set()
        
        # 判断当前执行的单元格类型
        if cell_id in self.processed_cell_ids:
            print(f"单元格 {cell_id}：这是编辑后重新执行的操作")
            # 在这里编写编辑后的业务逻辑
        else:
            print(f"单元格 {cell_id}：这是首次执行的新单元格")
            # 在这里编写新单元格的业务逻辑
            self.processed_cell_ids.add(cell_id)

# 将魔法命令注册到IPython环境
get_ipython().register_magics(CellTrackerMagics)

额外补充两点注意事项：

• cell_id是Jupyter环境特有的字段，如果在纯IPython终端运行，parent_header里可能没有这个值，所以一定要加判断避免报错；
• 我们用类属性processed_cell_ids来记录已处理的单元格，这样就能轻松区分首次执行和编辑后重复执行的场景。

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