Aquila-Consortium/csiborgtools
0
0
Fork 0
mirror of https://github.com/Richard-Sti/csiborgtools.git synced 2024-12-22 17:18:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update plots (#82)

Browse source
This commit is contained in:
Richard Stiskalek 2023-07-31 21:44:58 +02:00 committed by GitHub
parent 46848a86eb
commit 99d0232d7d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 23 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
scripts_plots/plot_data.py
View file

@ -136,8 +136,7 @@ def plot_hmf(pdf=False):
print("Plotting the HMF...", flush=True) print("Plotting the HMF...", flush=True)
paths = csiborgtools.read.Paths(**csiborgtools.paths_glamdring) paths = csiborgtools.read.Paths(**csiborgtools.paths_glamdring)
# csiborg_nsims = paths.get_ics("csiborg") csiborg_nsims = paths.get_ics("csiborg")
csiborg_nsims = [7444]
print("Loading CSiBORG halo counts.", flush=True) print("Loading CSiBORG halo counts.", flush=True)
for i, nsim in enumerate(tqdm(csiborg_nsims)): for i, nsim in enumerate(tqdm(csiborg_nsims)):
data = numpy.load(paths.halo_counts("csiborg", nsim)) data = numpy.load(paths.halo_counts("csiborg", nsim))
@ -163,16 +162,19 @@ def plot_hmf(pdf=False):
quijote_counts[i * nmax:(i + 1) * nmax, :] = data["counts"] quijote_counts[i * nmax:(i + 1) * nmax, :] = data["counts"]
quijote_counts /= numpy.diff(bins).reshape(1, -1) quijote_counts /= numpy.diff(bins).reshape(1, -1)
vol = 155.5**3
csiborg_counts /= vol
quijote_counts /= vol
x = 10**(0.5 * (bins[1:] + bins[:-1])) x = 10**(0.5 * (bins[1:] + bins[:-1]))
# Edit lower limits # Edit lower limits
csiborg_counts[:, x < 1e12] = numpy.nan csiborg_counts[:, x < 1e12] = numpy.nan
quijote_counts[:, x < 10**(12.4)] = numpy.nan quijote_counts[:, x < 10**(13.1)] = numpy.nan
# Edit upper limits # Edit upper limits
csiborg_counts[:, x > 4e15] = numpy.nan csiborg_counts[:, x > 3e15] = numpy.nan
quijote_counts[:, x > 4e15] = numpy.nan quijote_counts[:, x > 3e15] = numpy.nan
with plt.style.context(plt_utils.mplstyle): with plt.style.context(plt_utils.mplstyle):
cols = plt.rcParams["axes.prop_cycle"].by_key()["color"]
fig, ax = plt.subplots(nrows=2, sharex=True, fig, ax = plt.subplots(nrows=2, sharex=True,
figsize=(3.5, 2.625 * 1.25), figsize=(3.5, 2.625 * 1.25),
gridspec_kw={"height_ratios": [1, 0.65]}) gridspec_kw={"height_ratios": [1, 0.65]})
@ -194,20 +196,23 @@ def plot_hmf(pdf=False):
log_y = numpy.log10(mean_csiborg / mean_quijote) log_y = numpy.log10(mean_csiborg / mean_quijote)
err = numpy.sqrt((std_csiborg / mean_csiborg / numpy.log(10))**2 err = numpy.sqrt((std_csiborg / mean_csiborg / numpy.log(10))**2
+ (std_quijote / mean_quijote / numpy.log(10))**2) + (std_quijote / mean_quijote / numpy.log(10))**2)
ax[1].plot(x, 10**log_y, c=cols[2]) ax[1].plot(x, 10**log_y, c="gray")
ax[1].fill_between(x, 10**(log_y - err), 10**(log_y + err), alpha=0.5, ax[1].fill_between(x, 10**(log_y - err), 10**(log_y + err), alpha=0.5,
color=cols[2]) color="gray")
# Labels and accesories # Labels and accesories
ax[1].axhline(1, color="k", ls=plt.rcParams["lines.linestyle"], ax[1].axhline(1, color="k", ls="--",
lw=0.5 * plt.rcParams["lines.linewidth"], zorder=0) lw=0.5 * plt.rcParams["lines.linewidth"], zorder=0)
ax[0].set_ylabel(r"$\frac{\mathrm{d} n}{\mathrm{d}\log M_{\rm h}}~\mathrm{dex}^{-1}$") # noqa ax[0].set_ylabel(r"$\frac{\mathrm{d}^2 N}{\mathrm{d} V \mathrm{d}\log M_{\rm tot}}~[\mathrm{dex}^{-1} (\mathrm{Mpc} / h)^{-3}]$", # noqa
ax[1].set_xlabel(r"$M_{\rm h}~[M_\odot / h]$") fontsize="small")
ax[1].set_ylabel(r"$\mathrm{CSiBORG} / \mathrm{Quijote}$") ax[1].set_xlabel(r"$M_{\rm tot}~[M_\odot / h]$", fontsize="small")
ax[1].set_ylabel(r"$\mathrm{CSiBORG} / \mathrm{Quijote}$",
fontsize="small")
ax[0].set_xscale("log") ax[0].set_xscale("log")
ax[0].set_yscale("log") ax[0].set_yscale("log")
ax[1].set_yscale("log") ax[1].set_ylim(0.5, 2.0)
# ax[1].set_yscale("log")
ax[0].legend() ax[0].legend()
fig.tight_layout(h_pad=0, w_pad=0) fig.tight_layout(h_pad=0, w_pad=0)
@ -268,11 +273,11 @@ def plot_hmf_quijote_full(pdf=False):
ax[1].plot(x, counts[i, :] / mean, c=cols[0]) ax[1].plot(x, counts[i, :] / mean, c=cols[0])
# Labels and accesories # Labels and accesories
ax[1].axhline(1, color="k", ls=plt.rcParams["lines.linestyle"], ax[1].axhline(1, color="k", ls="--",
lw=0.5 * plt.rcParams["lines.linewidth"], zorder=0) lw=0.5 * plt.rcParams["lines.linewidth"], zorder=0)
ax[0].set_ylabel(r"$\frac{\mathrm{d}^2 n}{\mathrm{d}\log M_{\rm h} \mathrm{d} V}~[\mathrm{dex}^{-1} (\mathrm{Mpc / h})^{-3}]$", # noqa ax[0].set_ylabel(r"$\frac{\mathrm{d}^2 n}{\mathrm{d}\log M_{\rm tot} \mathrm{d} V}~[\mathrm{dex}^{-1} (\mathrm{Mpc / h})^{-3}]$", # noqa
fontsize="small") fontsize="small")
ax[1].set_xlabel(r"$M_{\rm h}~[$M_\odot / h]$", fontsize="small") ax[1].set_xlabel(r"$M_{\rm tot}~[$M_\odot / h]$", fontsize="small")
ax[1].set_ylabel(r"$\mathrm{HMF} / \langle \mathrm{HMF} \rangle$", ax[1].set_ylabel(r"$\mathrm{HMF} / \langle \mathrm{HMF} \rangle$",
fontsize="small") fontsize="small")
@ -635,10 +640,10 @@ if __name__ == "__main__":
if False: if False:
plot_mass_vs_ncells(7444, pdf=False) plot_mass_vs_ncells(7444, pdf=False)
if False: if True:
plot_hmf(pdf=False) plot_hmf(pdf=False)
if True: if False:
plot_hmf_quijote_full(pdf=False) plot_hmf_quijote_full(pdf=False)
if False: if False: