mirror of
https://github.com/balkian/gists.git
synced 2024-11-23 01:52:27 +00:00
28 lines
745 B
Python
28 lines
745 B
Python
|
"""
|
||
|
|
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
|