csiborgtools/notebooks/test_mmain.ipynb
Richard Stiskalek 1d847cbd06
Add Quijote (#61)
* Rename paths object

* Remove redshift calculation

* Explicit keywrod arg

* Rename box units

* Basic renaming

* Little docs

* Rename paths

* add imports

* Sort imports

* Add Quijote cat

* Split boxes

* add Quijote path

* Add origin argument

* Update nbs
2023-05-13 17:37:34 +01:00

6 KiB

In [1]:
import sys
import numpy as np
import matplotlib.pyplot as plt
import joblib
import scienceplots
sys.path.append("../")
import csiborgtools

plt.style.use(["science", "notebook"])
%matplotlib widget
%load_ext autoreload
%autoreload 2
In [2]:
d = np.load("/mnt/extraspace/rstiskalek/csiborg/split/clumps_07444_00951.npz")
In [12]:
%timeit d["232"]
772 µs ± 4.01 µs per loop (mean ± std. dev. of 7 runs, 1,000 loops each)
In [ ]:

In [2]:
paths = csiborgtools.read.Paths(**csiborgtools.paths_glamdring)
# cat = csiborgtools.read.ClumpsCatalogue(7444, paths)
In [3]:
paths.knn_path(7444, "auto", "la")
Out[3]:
'/mnt/extraspace/rstiskalek/csiborg/knn/auto/knncdf_07444_la.npz'
In [8]:
np.savez("test.npz", a=np.random.rand(510, 510, 510), b=np.random.rand(510, 510, 510))
In [15]:
d = np.load("test.npz")
In [16]:
d.files
Out[16]:
['a', 'b']
In [7]:
d.files
Out[7]:
['a', 'b']
In [ ]:

In [26]:
np.save("test.npy", np.array([]))
In [27]:
np.load("test.npy")
Out[27]:
array([], dtype=float64)
In [ ]:

In [ ]:

In [21]:
# paths.split_path(123, 7444, 952)
Out[21]:
'/mnt/extraspace/rstiskalek/csiborg/split/ic_00952/out_07444_123.npz'
In [4]:
nsim = 7444
nsnap = max(paths.get_snapshots(7444))
reader = csiborgtools.read.ParticleReader(paths)
In [14]:
# clumpind = reader.read_clumps(nsnap, nsim, cols="index")["index"]
In [ ]:

In [ ]:

In [17]:
parts = np.load("/mnt/extraspace/rstiskalek/csiborg/initmatch/clump_7468_particles.npy", allow_pickle=True)
In [ ]:

In [ ]: