def poly_surface_plot_imag(p, res=101): X, Y, vals = get_vals(p, (-1, 1), (-1, 1), res) fig = plt.figure() ax = fig.gca(projection='3d') ax.plot_surface(X, Y, vals.imag) plt.savefig("poly_surface_plot_imag.pdf")
def poly_color_plot_imag(p, res=101): X, Y, vals = get_vals(p, (-1, 1), (-1, 1), res=res) plt.pcolormesh(X, Y, vals.imag) plt.savefig("poly_color_plot_imag.pdf")
def poly_color_plot_real(p, res=101): X, Y, vals = get_vals(p, (-1, 1), (-1, 1), res=res) plt.pcolormesh(X, Y, vals.real) plt.savefig("poly_color_plot_real.pdf")