1237.sagews

def my_complex_plot(f, xrange, yrange, conts):
    p1 = complex_plot(f, xrange, yrange)
    p2 = contour_plot(lambda x,y: imag(f(x+y*I)), xrange, yrange, fill=False, cmap=[(0.5,0,0)],contours=conts)
    p3 = contour_plot(lambda x,y: imag(f(x+y*I)*I), xrange, yrange, fill=False, cmap=[(0.5,0,0)],contours=conts)
    return p1+p2+p3

my_complex_plot(lambda z: z^2, (-3,3), (-3,3), [-20,-19..20])

complex_plot(lambda z: z^2, (-3,3), (-3,3))

my_complex_plot(lambda z: sqrt(z), (-3,3), (-3,3), [-2,-1.9..2])

my_complex_plot(lambda z: exp(z), (-1,2), (-3.25,3.25), [-9,-8.5..9])

my_complex_plot(lambda z: log(z+0.001), (-3,3), (-3,3), [-3, -2.8..3])

x,y=var('x,y')

plot_vector_field((x,0), (x,-2,2), (y,-2,2), aspect_ratio=1)

plot_vector_field((0,x), (x,-2,2), (y,-2,2), aspect_ratio=1)

plot_vector_field((x,y), (x,-2,2), (y,-2,2), aspect_ratio=1)

plot_vector_field((-y,x), (x,-2,2), (y,-2,2), aspect_ratio=1)

t = var('t')

plot(1 + 0*t, (t, 0, 2*pi), ymin=-1, ymax=1)

plot(sin(t) + 0.5*cos(2*t) + 0.3*sin(3*t), (t, 0, 2*pi))

plot(-cos(3*t) + cos(4*t), (t, 0, 2*pi))

my_complex_plot(gamma, (-3,3), (-3,3), [-10..10])

my_complex_plot(zeta, (-3,3), (-5,5), [-10..10])