Bin mean return feature dtype

2019-05-28 16:58:36 +02:00 · 2019-05-28 16:58:36 +02:00 · f2c0c6e8d6
commit f2c0c6e8d6
parent b1a2d50a14
1 changed files with 2 additions and 1 deletions
--- a/idefix/vxl.py
+++ b/idefix/vxl.py
@ -131,7 +131,8 @@ def _bin_mean(grid, spatial, feature):
    density, edge = np.histogramdd(spatial, grid)
    weightd, edge = np.histogramdd(spatial, grid, weights=feature)
    mask = density == 0
-    return np.ma.masked_array(np.divide(weightd, density, where=~mask), mask)
+    return np.ma.masked_array(np.divide(weightd, density, where=~mask),
+                              mask, dtype=feature.dtype)

 def _bin_mode(grid, spatial, feature):
    '''Bin spatial in a grid, mode method.