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")