Python code

#!/usr/bin/env python

"""This generates four graphs for sin(z)."""
# Origin: Peter Halasz. 2010年09月14日
# Enhanced: Ika. 2013年07月23日
 
importnumpyasnp
frompylabimport *
frommpl_toolkits.mplot3dimport Axes3D
 
graphs = {'abs':abs, 'real':real, 'imag':imag, 'angle':angle}
 
for gr in graphs:
 ax = Axes3D(figure(), azim = -135, elev = 45)
 X = arange(-2*pi, 2*pi, pi/12)
 Y = arange(-2, 2, pi/12)
 X, Y = meshgrid(X, Y)
 fn = graphs[gr]
 Z = fn(sin(X + Y*1j)) # abs, real, imag, angle. angle range [-pi, pi]
 ax.plot_surface(X, Y, Z, rstride = 1, cstride = 1, cmap = cm.jet)
 ax.contour(X, Y, Z, zdir='z', offset=np.min(Z))
 ax.set_xlabel("x")
 ax.set_ylabel("y")
 ax.set_zlabel(gr + '(sin(x+iy))')
 plt.savefig("complex_sin_" + gr + "_01_Pengo.svg")