使用代码自定义 Jupyter Notebook 单元行为

Question

我正在 Jupyter Notebook 中编写和制作代码插图。我的用例是从某些代码单元中获取最终代码并将其放入 HTML 文档中。我找到了一个非常好的使用pygments包的管道，它为我突出显示代码并将其放入适当的 HTML 中。from pygments import highlightfrom pygments.lexers import PythonLexerfrom pygments.formatters import HtmlFormatterdef PyHighlight(code):    return highlight(code, PythonLexer(), HtmlFormatter())PyHighlight("print('Hello world!')")输出：'<div class="highlight"><pre><span></span><span class="nb">print</span><span class="p">(</span><span class="s1">&#39;Hello world!&#39;</span><span class="p">)</span>\n</pre></div>\n'但是将每个代码单元格转换成字符串然后传递给函数PyHighlight并最终得到 HTML 对我来说非常繁琐。有没有办法可以将每个单元格的内容作为字符串获取？更好的是，我可以PyHighlight在每个单元格内容作为参数的单元格之后触发运行，PyHightlight这样我就可以复制粘贴突出显示的代码 HTML 吗？

Answer 1

我从关于Custom Magic 的IPython 文档中获得了一些灵感

在单元格中定义并注册以下魔法。

from IPython.core.magic import  (Magics, cell_magic)

@magics_class

class MyMagics(Magics):

    @cell_magic

    def pygmented(self, line, cell):

        print(PyHighlight(cell))

get_ipython().register_magics(MyMagics)

之后，只需添加%%pygmented到每个单元格的顶部，运行单元格后，单元格的内容也会在所有突出显示后打印出来（如问题中所问）。