diff --git a/repos/61a2fae8d94ede667f4a/remove_output.py b/repos/61a2fae8d94ede667f4a/remove_output.py new file mode 100644 index 0000000..3fc2ea9 --- /dev/null +++ b/repos/61a2fae8d94ede667f4a/remove_output.py @@ -0,0 +1,28 @@ +""" +Usage: python remove_output.py notebook.ipynb [ > without_output.ipynb ] +Modified from remove_output by Minrk + +""" +import sys +import io +import os +from IPython.nbformat.current import read, write + + +def remove_outputs(nb): + """remove the outputs from a notebook""" + for ws in nb.worksheets: + for cell in ws.cells: + if cell.cell_type == 'code': + cell.outputs = [] + +if __name__ == '__main__': + fname = sys.argv[1] + with io.open(fname, 'r') as f: + nb = read(f, 'json') + remove_outputs(nb) + base, ext = os.path.splitext(fname) + new_ipynb = "%s_removed%s" % (base, ext) + with io.open(new_ipynb, 'w', encoding='utf8') as f: + write(nb, f, 'json') + print "wrote %s" % new_ipynb \ No newline at end of file