Question

>> masks = [[1,1],[0,0]]    
>> [np.ma.masked_array(data=np.array([1.0,2.0]), mask=m, fill_value=np.nan).mean() for m in masks]
   [masked, 1.5]

我想用masked代替nan结果。有没有一种方法可以直接使用numpy的masked_array？

Answer 1

In [232]: M = np.ma.masked_array(data=np.array([1.0,2.0]),mask=[True, False])

filled方法将填充值替换为掩码值：

In [233]: M.filled()                                                         
Out[233]: array([1.e+20, 2.e+00])
In [234]: M.filled(np.nan)         # or with a value of your choice.                                                   
Out[234]: array([nan,  2.])

或者您也可以在定义数组时指定填充值：

In [235]: M = np.ma.masked_array(data=np.array([1.0,2.0]),mask=[True, False],
     ...:  fill_value=np.nan)                                                
In [236]: M                                                                  
Out[236]: 
masked_array(data=[--, 2.0],
             mask=[ True, False],
       fill_value=nan)
In [237]: M.filled()                                                         
Out[237]: array([nan,  2.])

被掩盖的均值方法跳过填充值：

In [238]: M.mean()                                                           
Out[238]: 2.0
In [239]: M.filled().mean()                                                  
Out[239]: nan
In [241]: np.nanmean(M.filled())    # so does the `nanmean` function
In [242]: M.data.mean()             # mean of the underlying data                                                      
Out[242]: 1.5

Answer 2

我认为您可以np.ones

masks=np.array([[1,1],[0,0]])

np.ma.masked_array(data=np.array([1.0,2.0])*np.ones(masks.shape), mask=masks, fill_value=np.nan).mean(axis=1)
Out[145]: 
masked_array(data=[--, 1.5],
             mask=[ True, False],
       fill_value=1e+20)

用numpy的masked_array中的nan替换masked

2 个答案: