diff --git a/csiborgtools/read/summaries.py b/csiborgtools/read/summaries.py index b2163ee..6222b4f 100644 --- a/csiborgtools/read/summaries.py +++ b/csiborgtools/read/summaries.py @@ -18,6 +18,7 @@ Tools for summarising various results. import numpy import joblib from os.path import (join, isfile) +from glob import glob from tqdm import tqdm @@ -283,6 +284,25 @@ class kNNCDFReader: cdf = cdf[0, ...] if nknns == 1 else cdf # Reshape if necessary return cdf + @staticmethod + def cross_files(ic, folder): + """ + Return the file paths corresponding to the cross-correlation of a given + IC. + + Parameters + ---------- + ic : int + The desired IC. + folder : str + The folder containing the cross-correlation files. + + Returns + ------- + filepath : list of str + """ + return [file for file in glob(join(folder, "*")) if str(ic) in file] + class PairOverlap: r""" diff --git a/meetings/220403_knn.ipynb b/meetings/220403_knn.ipynb index da4822f..b5d2d9d 100644 --- a/meetings/220403_knn.ipynb +++ b/meetings/220403_knn.ipynb @@ -6,8 +6,8 @@ "id": "5a38ed25", "metadata": { "ExecuteTime": { - "end_time": "2023-04-02T19:21:33.573666Z", - "start_time": "2023-04-02T19:21:30.143966Z" + "end_time": "2023-04-02T21:16:58.830606Z", + "start_time": "2023-04-02T21:16:55.025438Z" }, "scrolled": true }, @@ -42,12 +42,12 @@ }, { "cell_type": "code", - "execution_count": 4, + "execution_count": 13, "id": "6083fcbe", "metadata": { "ExecuteTime": { - "end_time": "2023-04-02T19:23:31.324449Z", - "start_time": "2023-04-02T19:23:29.114870Z" + "end_time": "2023-04-02T21:20:09.950356Z", + "start_time": "2023-04-02T21:20:06.317948Z" } }, "outputs": [ @@ -55,7 +55,7 @@ "name": "stderr", "output_type": "stream", "text": [ - "100%|██████████| 101/101 [00:01<00:00, 73.91it/s]\n" + "100%|██████████| 101/101 [00:02<00:00, 38.07it/s]\n" ] } ], @@ -69,12 +69,12 @@ }, { "cell_type": "code", - "execution_count": 5, + "execution_count": 14, "id": "6ec8e4a5", "metadata": { "ExecuteTime": { - "end_time": "2023-04-02T19:23:33.894435Z", - "start_time": "2023-04-02T19:23:32.073212Z" + "end_time": "2023-04-02T21:20:11.664874Z", + "start_time": "2023-04-02T21:20:09.952179Z" }, "scrolled": false }, @@ -1047,7 +1047,7 @@ { "data": { "text/html": [ - "" + "" ], "text/plain": [ "" @@ -1086,12 +1086,12 @@ }, { "cell_type": "code", - "execution_count": 7, + "execution_count": 16, "id": "1f9898f6", "metadata": { "ExecuteTime": { - "end_time": "2023-04-02T19:23:40.416015Z", - "start_time": "2023-04-02T19:23:39.223369Z" + "end_time": "2023-04-02T21:21:19.279253Z", + "start_time": "2023-04-02T21:21:16.309685Z" } }, "outputs": [ @@ -1099,7 +1099,7 @@ "name": "stderr", "output_type": "stream", "text": [ - "100%|██████████| 35/35 [00:01<00:00, 32.52it/s]\n" + "100%|██████████| 100/100 [00:02<00:00, 42.21it/s]\n" ] } ], @@ -1107,7 +1107,8 @@ "knnreader = csiborgtools.read.kNNCDFReader()\n", "\n", "cross_files = glob(\"/mnt/extraspace/rstiskalek/csiborg/knn/cross/knncdf_9004*\")\n", - "len(cross_files)\n", + "\n", + "cross_files = knnreader.cross_files(7444, \"/mnt/extraspace/rstiskalek/csiborg/knn/cross/\")\n", "\n", "ks = [0, 3, 7]\n", "rs, corr, mass_thresholds = knnreader.read(cross_files, ks, rmin=1)" @@ -1115,12 +1116,12 @@ }, { "cell_type": "code", - "execution_count": 10, + "execution_count": 17, "id": "e0fcf77b", "metadata": { "ExecuteTime": { - "end_time": "2023-04-02T19:24:34.981711Z", - "start_time": "2023-04-02T19:24:31.703199Z" + "end_time": "2023-04-02T21:21:22.690776Z", + "start_time": "2023-04-02T21:21:19.980499Z" } }, "outputs": [ @@ -2092,7 +2093,7 @@ { "data": { "text/html": [ - "" + "" ], "text/plain": [ ""