csiborgtools/notebooks/powerspectrum_test.ipynb
Richard Stiskalek aaa14fc880
Add density field plot and start preparing CSiBORG2 (#94)
* Add RAMSES2HDF5 conversion

* Upload changes

* Clean up

* More clean up

* updates

* Little change

* pep9

* Add basic SPH calculation for a snapshot

* Add submit script

* Remove echo

* Little changes

* Send off changes

* Little formatting

* Little updates

* Add nthreads argument

* Upload chagnes

* Add nthreads arguemnts

* Some local changes..

* Update scripts

* Add submission script

* Update script

* Update params

* Rename CSiBORGBox to CSiBORG1box

* Rename CSiBORG1 reader

* Move files

* Rename folder again

* Add basic plotting here

* Add new skeletons

* Move def

* Update nbs

* Edit directories

* Rename files

* Add units to converted snapshots

* Fix empty dataset bug

* Delete file

* Edits to submission scripts

* Edit paths

* Update .gitignore

* Fix attrs

* Update weighting

* Fix RA/dec bug

* Add FORNAX cluster

* Little edit

* Remove boxes since will no longer need

* Move func back

* Edit to include sort by membership

* Edit paths

* Purge basic things

* Start removing

* Bring file back

* Scratch

* Update the rest

* Improve the entire file

* Remove old things

* Remove old

* Delete old things

* Fully updates

* Rename file

* Edit submit script

* Little things

* Add print statement

* Add here cols_to_structured

* Edit halo cat

* Remove old import

* Add comment

* Update paths manager

* Move file

* Remove file

* Add chains
2023-12-13 16:08:25 +00:00

1,009 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.CSiBORGPaths(**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 [ ]: