A python script to parse commands and outputs from jupyter notebook

jupyter_notebook_output_parser.py

import json

nb_fname = "xxx.ipynb"
with open(nb_fname, "r", encoding="utf-8") as f:
    nb = json.load(f)
cells = nb['cells']
print("There are", len(cells), "cells")

for cell in cells:
    source = ''.join(cell['source'])
    print(source)
    outputs = cell['outputs']
    if len(outputs) == 0:
        continue
    try:
        text = ''.join(outputs[0]['data']['text/plain'])
    except:
        text = ''.join(outputs[0]['text'])
    print(text)
    print("===========================")