Aquila-Consortium/csiborgtools
0
0
Fork 0
mirror of https://github.com/Richard-Sti/csiborgtools.git synced 2024-10-19 00:25:05 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity
c14be720b5
csiborgtools/notebooks/fits.ipynb
Richard Stiskalek c14be720b5
Particle dumping (#49) 
* Add paths

* Add dumping script

* Remove snap from density fields

* Add progress statement

* Update import ordering

* Update nb
2023-04-28 13:57:23 +01:00

3.2 KiB
Raw Blame History

In [1]:
import sys
import numpy as np
import matplotlib.pyplot as plt

sys.path.append("../")
import csiborgtools

%matplotlib widget 
%load_ext autoreload
%autoreload 2
In [2]:
paths = csiborgtools.read.CSiBORGPaths(**csiborgtools.paths_glamdring)
nsim0 = 7444
nsimx = 7444 + 24
nsnap0 = max(paths.get_snapshots(nsim0))
nsnapx = max(paths.get_snapshots(nsimx))
overlapper  = csiborgtools.match.ParticleOverlap()
In [3]:
halo0_archive = np.load(paths.split_path(nsnap0, nsim0))
halox_archive = np.load(paths.split_path(nsnapx, nsimx))
In [13]:
# delta_bckg = overlapper.make_bckg_delta(halo0_archive, verbose=True)
# delta_bckg = overlapper.make_bckg_delta(halox_archive, delta=delta_bckg, verbose=True)
# np.save("./bckg_{}_{}.npy".format(nsim0, nsimx), delta_bckg)

100%|██████████| 204873/204873 [16:48<00:00, 203.15it/s]
100%|██████████| 204960/204960 [16:42<00:00, 204.55it/s]
In [1]:
import h5py
In [2]:
f = h5py.File("../data/particles_7444.h5", "r")
In [8]:
f["particles"][0, :]
Out[8]:
array([ 4.42346358e-01,  7.09757663e-03,  4.64053304e-01, -1.96926287e-03,
       -2.67177823e-03, -6.45721859e-04,  1.16415322e-10])
In [ ]: