复数的实部和虚部的等值线图

产生一个复数$z$, $z$的实部和虚部分别从-100变化到100，在一幅图的两父子图中分别$z^2$的实部和虚部的等值线图.

import matplotlib.pyplot as plt 
import numpy as np

x_vals=np.linspace(-100,100,250)         #创建自变量数组
y_vals=np.linspace(-100,100,250)             #创建自变量数组

x,y=np.meshgrid(x_vals,y_vals)            #将原始数据变成网格数据形式
z=x+1j*y
z_sq=z**2
plt.figure()
#分图
plt.subplot(1,2,1)
cs1=plt.contour(x,y,z_sq.real)        #返回值coutourset 对象
plt.clabel(cs1,fontsize=10)      #等高线贴标签
plt.subplot(1,2,2)
cs2=plt.contour(x,y,z_sq.imag)          #返回值coutourset 对象
plt.clabel(cs2,fontsize=10)      #等高线贴标签

结果：