florent/idefix
florent/idefix
1
1
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

Bin mean return feature dtype

Browse Source
This commit is contained in:
Florent Guiotte 2019-05-28 16:58:36 +02:00
parent b1a2d50a14
commit f2c0c6e8d6
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
idefix/vxl.py
View File

@ -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.