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

Add better colorbar

Browse source
This commit is contained in:
rstiskalek 2023-06-06 08:41:39 +01:00
parent 70b173ccc5
commit 004d9629a2
1 changed files with 16 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
scripts_plots/plot_data.py
View file

@ -190,17 +190,29 @@ def plot_projected_field(kind, nsim, grid, in_rsp, MAS="PCS",
if highres_only: if highres_only:
csiborgtools.field.fill_outside(field, numpy.nan, rmax=155.5, csiborgtools.field.fill_outside(field, numpy.nan, rmax=155.5,
boxsize=677.7) boxsize=677.7)
start = field.shape[0] // 4 # start = field.shape[0] // 4
end = field.shape[0] - start start = round(field.shape[0] * 0.27)
end = round(field.shape[0] * 0.73)
# end = field.shape[0] - start
field = field[start:end, start:end, start:end] field = field[start:end, start:end, start:end]
labels = [r"$y-z$", r"$x-z$", r"$x-y$"]
with plt.style.context(utils.mplstyle): with plt.style.context(utils.mplstyle):
fig, ax = plt.subplots(figsize=(3.5 * 2, 2.625), ncols=3, sharey=True, fig, ax = plt.subplots(figsize=(3.5 * 2, 2.625), ncols=3, sharey=True,
sharex=True) sharex=True)
fig.subplots_adjust(hspace=0, wspace=0) fig.subplots_adjust(hspace=0, wspace=0)
for i in range(3): for i in range(3):
im = ax[i].imshow(numpy.nanmean(field, axis=i)) img = numpy.nanmean(field, axis=i)
fig.colorbar(im) if i == 0:
vmin, vmax = numpy.nanpercentile(img, [1, 99])
im = ax[i].imshow(numpy.nanmean(field, axis=i), vmin=vmin,
vmax=vmax)
else:
ax[i].imshow(numpy.nanmean(field, axis=i), vmin=vmin,
vmax=vmax)
ax[i].set_title(labels[i])
cbar_ax = fig.add_axes([1.0, 0.1, 0.025, 0.8])
fig.colorbar(im, cax=cbar_ax, label="Mean projected field")
fig.tight_layout(h_pad=0, w_pad=0) fig.tight_layout(h_pad=0, w_pad=0)
for ext in ["png"] if pdf is False else ["png", "pdf"]: for ext in ["png"] if pdf is False else ["png", "pdf"]: