Python script that clears all output cells from given notebook

ipynb_clear_output_cells.py

import argparse
from nbformat import read, write, NO_CONVERT


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("notebook",  help="path to jupyter notebook")
    parser.add_argument("-o", "--output",  default="processed.ipynb", help="output path")
    args = parser.parse_args()

    src = args.notebook
    output = args.output

    print("Loading notebook...")
    with open(src, 'r') as f:
        nb = read(f, NO_CONVERT)

    print("Removing all output...")
    out_cells = filter(lambda c: c.cell_type == 'code', nb.cells)
    for cell in out_cells:
        cell.outputs = []

    print("Saving to %s..." % output)
    with open(output, 'w', encoding='utf8') as f:
        write(nb, f, NO_CONVERT)

    print("Completed...")

if __name__ == '__main__':
    main()