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

Update name

Browse source
This commit is contained in:
rstiskalek 2024-09-23 10:23:33 +01:00
parent da1b317861
commit 8b92d632aa
1 changed files with 13 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
notebooks/flow/reconstruction_comparison.py
View file

@ -59,6 +59,7 @@ def names_to_latex(names, for_corner=False):
"mag_cal": "\\mathcal{M}",
"l": "\\ell ~ [\\mathrm{deg}]",
"b": "b ~ [\\mathrm{deg}]",
"rLG": "R_{\\rm offset} ~ [\\mathrm{Mpc} / h]",
}
ltx_corner = {"alpha": r"$\alpha$",
@ -209,17 +210,18 @@ def get_samples(fname, convert_Vext_to_galactic=True):
for key in grp.keys():
samples[key] = grp[key][...]
if convert_Vext_to_galactic:
Vext = samples.pop("Vext")
samples["Vmag"] = np.linalg.norm(Vext, axis=1)
Vext = csiborgtools.cartesian_to_radec(Vext)
samples["l"], samples["b"] = csiborgtools.radec_to_galactic(
Vext[:, 1], Vext[:, 2])
else:
Vext = samples.pop("Vext")
samples["Vx"] = Vext[:, 0]
samples["Vy"] = Vext[:, 1]
samples["Vz"] = Vext[:, 2]
if "Vext" in samples:
if convert_Vext_to_galactic:
Vext = samples.pop("Vext")
samples["Vmag"] = np.linalg.norm(Vext, axis=1)
Vext = csiborgtools.cartesian_to_radec(Vext)
samples["l"], samples["b"] = csiborgtools.radec_to_galactic(
Vext[:, 1], Vext[:, 2])
else:
Vext = samples.pop("Vext")
samples["Vx"] = Vext[:, 0]
samples["Vy"] = Vext[:, 1]
samples["Vz"] = Vext[:, 2]
keys = list(samples.keys())
for key in keys: