如何为 Jupyter 笔记本中的每个单元启用计时魔法？

Question

%%time 和 %%timeit 魔法可以在 Jupyter 或 iPython 笔记本中对单个单元格进行计时。

是否有类似的功能可以为 Jupyter 笔记本中的每个单元打开和关闭计时？

This question 是相关的，但对于在每个单元格中自动启用给定魔法这一更普遍的问题没有答案。

Answer 1

通过 custom.js 文件（通常放在~/.jupyter/custom/custom.js）来实现这一点的一个hacky方法

如何为工具栏创建按钮的示例位于here，这是我此答案的依据。它只是在按下启用按钮时将您想要的魔法的字符串形式添加到所有单元格，禁用按钮使用str.replace 将其“关闭”。

define([
    'base/js/namespace',
    'base/js/events'
], function(Jupyter, events) {
    events.on('app_initialized.NotebookApp', function(){
        Jupyter.toolbar.add_buttons_group([
            {
                'label'   : 'enable timing for all cells',
                'icon'    : 'fa-clock-o', // select your icon from http://fortawesome.github.io/Font-Awesome/icons
                'callback': function () {
                    var cells = Jupyter.notebook.get_cells();
                    cells.forEach(function(cell) {
                        var prev_text = cell.get_text();
                        if(prev_text.indexOf('%%time\n%%timeit\n') === -1) {
                            var text  = '%%time\n%%timeit\n' + prev_text;
                            cell.set_text(text);
                        }
                    });
                }
            },
            {
                'label'   : 'disable timing for all cells',
                'icon'    : 'fa-stop-circle-o', // select your icon from http://fortawesome.github.io/Font-Awesome/icons
                'callback': function () {
                    var cells = Jupyter.notebook.get_cells();
                    cells.forEach(function(cell) {
                        var prev_text = cell.get_text();
                        var text  = prev_text.replace('%%time\n%%timeit\n','');
                        cell.set_text(text);
                    });
                }
            }
            // add more button here if needed.
        ]);
    });
});