Skip to content

Instantly share code, notes, and snippets.

@andreipit
Forked from damianavila/remove_output.py
Last active May 29, 2019 10:40
Show Gist options
  • Save andreipit/e89887582b67ec428c1aba175c7979a2 to your computer and use it in GitHub Desktop.
Save andreipit/e89887582b67ec428c1aba175c7979a2 to your computer and use it in GitHub Desktop.
Fix of print function for python 3
"""
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)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment