From 380db96bc3998f6c0ab44ecd5d362c39d518f899 Mon Sep 17 00:00:00 2001 From: Pierre-Francois Leget Date: Fri, 28 Feb 2025 14:11:31 -0500 Subject: [PATCH 1/2] return bin edge --- treegp/meanify.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/treegp/meanify.py b/treegp/meanify.py index 61cf39b..8905497 100644 --- a/treegp/meanify.py +++ b/treegp/meanify.py @@ -5,6 +5,7 @@ import numpy as np from scipy.stats import binned_statistic_2d import fitsio +import copy class meanify(object): @@ -97,7 +98,7 @@ def meanify(self, lu_min=None, lu_max=None, lv_min=None, lv_max=None): ) wrms = np.sqrt(wvar) else: - average, u0, v0, bin_target = binned_statistic_2d( + average, xedge, yedge, bin_target = binned_statistic_2d( coords[:, 0], coords[:, 1], params, @@ -113,7 +114,9 @@ def meanify(self, lu_min=None, lu_max=None, lv_min=None, lv_max=None): wrms = wrms.reshape(-1) Filter &= np.isfinite(average).reshape(-1) Filter &= np.isfinite(wrms).reshape(-1) - params0 = average + params0 = copy.deepcopy(average) + u0 = copy.deepcopy(xedge) + v0 = copy.deepcopy(yedge) wrms0 = wrms # get center of each bin @@ -122,6 +125,9 @@ def meanify(self, lu_min=None, lu_max=None, lv_min=None, lv_max=None): u0, v0 = np.meshgrid(u0, v0) self._u0 = u0 self._v0 = v0 + self._xedge = xedge + self._yedge = yedge + self._average = average coords0 = np.array([u0.reshape(-1), v0.reshape(-1)]).T From 57e93faf988a1170f87089cd9dc0740eeb63f1b3 Mon Sep 17 00:00:00 2001 From: Pierre-Francois Leget Date: Fri, 28 Feb 2025 14:23:33 -0500 Subject: [PATCH 2/2] typo --- treegp/meanify.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/treegp/meanify.py b/treegp/meanify.py index 8905497..4f063a5 100644 --- a/treegp/meanify.py +++ b/treegp/meanify.py @@ -108,7 +108,7 @@ def meanify(self, lu_min=None, lu_max=None, lv_min=None, lv_max=None): wrms = np.zeros_like(average) average = average.T wrms = wrms.T - self._average = average + self._average = copy.deepcopy(average) self._wrms = wrms average = average.reshape(-1) wrms = wrms.reshape(-1) @@ -127,7 +127,6 @@ def meanify(self, lu_min=None, lu_max=None, lv_min=None, lv_max=None): self._v0 = v0 self._xedge = xedge self._yedge = yedge - self._average = average coords0 = np.array([u0.reshape(-1), v0.reshape(-1)]).T