import matplotlib.pyplot as plt
import numpy as np

m = 1
k = 1
gamma = 0.1

omega = np.linspace(-2, 2, 10000)
A = 1/(-omega*omega*m + k + omega*1j*gamma)

#plot magnitude against omega
plt.plot(omega, np.absolute(A))

plt.ylabel('magnitude')
plt.xlabel('omega')
plt.show()

#plot phase against omega
plt.plot(omega, np.angle(A))

plt.ylabel('phase')
plt.xlabel('omega')
plt.show()