Simple Complexity chart (IPython notebook & matplotlib)

plots.py

import math
import matplotlib.pyplot as plt
%matplotlib inline

a = range(1,100)
b = map(lambda x:x*math.log(x,2), a)
c = map(lambda x:x*x, a)
d = map(lambda x:2**x, a)

p = plt.figure()
sbp = p.add_subplot(1,1,1)
sbp.plot(a,b,'r', label=r'$n\log n$')
sbp.plot(a,a,'b', label=r'$n$')
sbp.plot(a,c,'c', label=r'$n^2$')
sbp.plot(a,d,'m', label=r'$2^n$')
sbp.set_ylim([0, 1000])
sbp.legend(loc=1)
plt.show()