pp.ylabel("green")
pp.colorbar()
pp.contour(p, levels=np.linspace(-10, p.max(), 15), origin="lower")
f.axes[0].set_title("Log-Density")
f.savefig("2D_ex_logdense.pdf")

f = pp.figure()
pp.contour(p, levels=np.linspace(-10, p.max(), 15), origin="lower")
pp.imshow(np.ma.masked_where(c.labels <= 0, c.labels), origin="lower", alpha=0.4, interpolation="nearest")
pp.xlabel("blue")
pp.ylabel("green")
f.axes[0].set_title("Initial Modes")
f.savefig("2D_ex_initial_labels.pdf")

f = pp.figure()
c = ms.persistence_merge(0.5)
pp.contour(p, levels=np.linspace(-10, p.max(), 15), origin="lower")
pp.imshow(np.ma.masked_where(c.labels <= 0, c.labels), origin="lower", alpha=0.4, interpolation="nearest")
pp.xlabel("blue")
pp.ylabel("green")
f.axes[0].set_title("Persistent Modes")
f.savefig("2D_ex_persistent_labels.pdf")

f = pp.figure()
c.refine_labels()
pp.contour(p, levels=np.linspace(-10, p.max(), 15), origin="lower")
pp.imshow(np.ma.masked_where((c.labels <= 0), c.labels), origin="lower", alpha=0.4, interpolation="nearest")
pp.xlabel("blue")
pp.ylabel("green")
f.axes[0].set_title("Partially Resolved Boundaries")
f.savefig("2D_ex_resolved_labels.pdf")