from mpl_toolkits.mplot3d import Axes3D
import matplotlib.cm as cm
def s_plane_plot(sfunc, limits = [3,3,10], nsamp = 500):
fig = plt.figure()
ax = fig.gca(projection = '3d')
sigma = np.linspace(-limits[0], limits[0], nsamp)
omega = sigma.copy()
sigma, omega = np.meshgrid(sigma, omega)
s = sigma + 1j*omega
surf = ax.plot_surface(sigma, omega, np.abs(sfunc(s)), cmap = cm.flag)
ax.set_zlim(0, limits[2])
plt.xlabel('$sigma$')
plt.ylabel('$jomega$')
fig.tight_layout()
def X(s): return 1/((s + .2+.5j)*(s + .2-.5j))
s_plane_plot(X, limits = [1,1,4], nsamp =40)
sympy.Heaviside(t-2)
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.cm as cm
def s_plane_plot(sfunc, limits = [3,3,10], nsamp = 500):
fig = plt.figure()
ax = fig.gca(projection = '3d')
sigma = np.linspace(-limits[0], limits[0], nsamp)
omega = sigma.copy()
sigma, omega = np.meshgrid(sigma, omega)
s = sigma + 1j*omega
surf = ax.plot_surface(sigma, omega, np.abs(sfunc(s)), cmap = cm.flag)
ax.set_zlim(0, limits[2])
plt.xlabel('$sigma$')
plt.ylabel('$jomega$')
fig.tight_layout()
def X(s): return 1/((s + .2+.5j)*(s + .2-.5j))
s_plane_plot(X, limits = [1,1,4], nsamp =40)
sympy.Heaviside(t-2)