In [1]:
%matplotlib notebook
import numpy as np
import matplotlib.pyplot as plt
In [12]:
w = np.linspace(0,2,1000)
m=1
k=1
gamma=.1
A=1/(-m*w*w+gamma*1j*w+k)
fig=plt.figure()
#Since amplitude cannot be negative, take the absolute value
plt.plot(w,np.absolute(A))
plt.xlabel('frequency')
plt.ylabel('ampitude')
plt.show()
plt.figure()
plt.plot(w,np.angle(A))
plt.xlabel('frequency')
plt.ylabel('phase')
plt.show()
In [ ]: