memprof.py

import gc
import os
import resource
import sys
import time

def memusage():
    """memory usage in MB. getrusage defaults to KB on Linux."""
    return resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / 1e3

def run():
    print("Python %s" % sys.version)
    print("Memory usage at start: %.2f MB" % memusage())
    t_start = time.time()
    for i in range(4000000):
        pass
    print("spent time: {:.2f}".format(time.time() - t_start))
    print("Memory usage after loop: %.2f MB" % memusage())
    gc.collect()
    print("Memory usage after GC: %.2f MB" % memusage())

if __name__ == "__main__":
    run()