diff --git a/plot_grav_profs.py b/plot_grav_profs.py new file mode 100644 index 0000000..bdbd8ab --- /dev/null +++ b/plot_grav_profs.py @@ -0,0 +1,29 @@ +# coding: utf-8 +import unyt as u +import numpy as np +import matplotlib.pyplot as plt +from calc_toomre import DiskModel + +r = np.linspace(1,201) * u.kpc + +dsk = DiskModel() + +g_nfw_mn = dsk.g(r) +g_modnfw = dsk.g_modNFW(r) + +fig, ax = plt.subplots(ncols=2) + +ax[0].plot(r, np.sqrt(r*g_nfw_mn).to('km/s'), label='NFW+MN') +ax[0].plot(r, np.sqrt(r*g_modnfw).to('km/s'), label='modNFW') + +ax[1].plot(r, g_nfw_mn.to('km/s**2')) +ax[1].plot(r, g_modnfw.to('km/s**2')) + +ax[0].set_xlabel("r [kpc]") +ax[1].set_xlabel("r [kpc]") + +ax[0].set_ylabel(r"$v_\mathrm{circ}$ [km/s]") +ax[1].set_ylabel(r"g [$\mathrm{km/s^2}$]") + +ax[0].legend() +plt.show() diff --git a/calc_toomre.py b/toy_disk.py similarity index 100% rename from calc_toomre.py rename to toy_disk.py