将参数传递给jupyter笔记本以nbformat开头

时间:2017-08-08 10:16:55

标签: python python-3.x jupyter-notebook jupyter papermill

我是 Jupyter 的初学者。我有一个Python程序,可以使用nbformat execute API启动我的笔记本。

它运作良好,但有一件事我没有想到; 如何将数据从我的程序传递到我即将执行的笔记本?

为了完整性,这里是我用来运行笔记本的代码:

import nbformat
from nbconvert.preprocessors import ExecutePreprocessor

class NotebookExecutor:
    def __init__(self, name, base_path, notebook_filename_in, notebook_filename_out, timeout=-1):
        self.name = name
        if base_path.endswith('/') is False:
            base_path = base_path + '/'
        self.base_path = base_path
        self.notebook_filename_in = notebook_filename_in
        self.notebook_filename_out = notebook_filename_out
        self.timeout = timeout

    def run(self):
        print("Running notebook '" + self.name + "'")
        nb = nbformat.read(open(self.base_path + self.notebook_filename_in), as_version=4)
        ep = ExecutePreprocessor(timeout=self.timeout, kernel_name='python3', allow_errors=True)
        try:
            ep.preprocess(nb, {'metadata': {'path': self.base_path}})
        except CellExecutionError:
            msg = 'Error executing the notebook "%s".\n\n' % self.notebook_filename_in
            msg += 'See notebook "%s" for the traceback.' % self.notebook_filename_out
            print(msg)
        # raise
        finally:
            nbformat.write(nb, open(self.base_path + self.notebook_filename_out, mode='wt'))

ne = NotebookExecutor('Test', '/my/path', 'MyBook.ipynb', 'MyBook_out.ipynb')
ne.run()

1 个答案:

答案 0 :(得分:1)

我建议使用papermill,我不确定这是否有助于您解决问题。

https://github.com/nteract/papermill

我可以使用它来循环使用不同参数的笔记本。