3

我想从压缩的掩码数组和相应的掩码中创建一个数组。用一个例子更容易解释这一点：

>>> x=np.ma.array(np.arange(4).reshape((2,2)), mask = [[True,True],[False,False]])
>>> y=x.compressed()
>>> y
array([ 2,  3])

现在我想创建一个与 x 形状相同的数组，其中掩码值获得标准值（例如 -1），其余部分填充给定数组。它应该像这样工作：

>>> z = decompress(y, mask=[[True,True],[False,False]], default=-1)
>>> z
array([[-1, -1],
       [ 2,  3]])

问题是：有没有像“解压缩”这样的方法，还是我需要自己编码？在 Fortran 中，这是通过“pack”和“unpack”方法完成的。感谢您的任何建议。

4

1 回答 1

6

虽然我已经回答了一些ma问题，但我绝不是它的专家。但我会探讨这个问题

让我们概括一下您的 a 数组：

In [934]: x=np.ma.array(np.arange(6).reshape((2,3)), mask = [[True,True,False],[False,False,True]])
In [935]: x
Out[935]: 
masked_array(data =
 [[-- -- 2]
 [3 4 --]],
             mask =
 [[ True  True False]
 [False False  True]],
       fill_value = 999999)
In [936]: y=x.compressed()
In [937]: y
Out[937]: array([2, 3, 4])

y除了值的子集之外没有任何信息x。注意是1d

x将其值存储在 2 个数组中（实际上这些是访问底层._data,属性的._mask属性）：

In [938]: x.data
Out[938]: 
array([[0, 1, 2],
       [3, 4, 5]])
In [939]: x.mask
Out[939]: 
array([[ True,  True, False],
       [False, False,  True]], dtype=bool)

我的猜测是，de-compress我们需要empty用正确的 dtype、shape 和 mask 制作一个掩码数组，并将 的值复制y到它的data. 但是应该将什么值放入被屏蔽的元素中data？

或者另一种解决问题的方法 - 是否可以将值从y后面复制到x?

一种可能的解决方案是将新值复制到x[~x.mask]：

In [957]: z=2*y
In [958]: z
Out[958]: array([4, 6, 8])
In [959]: x[~x.mask]=z
In [960]: x
Out[960]: 
masked_array(data =
 [[-- -- 4]
 [6 8 --]],
             mask =
 [[ True  True False]
 [False False  True]],
       fill_value = 999999)
In [961]: x.data
Out[961]: 
array([[0, 1, 4],
       [6, 8, 5]])

或者制作一个新数组

In [975]: w=np.zeros_like(x)
In [976]: w[~w.mask]=y
In [977]: w
Out[977]: 
masked_array(data =
 [[-- -- 2]
 [3 4 --]],
             mask =
 [[ True  True False]
 [False False  True]],
       fill_value = 999999)
In [978]: w.data
Out[978]: 
array([[0, 0, 2],
       [3, 4, 0]])

另一种方法是制作一个full带有无效值的常规数组，y像这样复制，然后将整个东西变成一个掩码数组。可能有一个掩码数组构造函数，它允许您仅指定有效值和掩码。但我必须为此深入研究文档。

================

将执行此操作的另一个操作序列，np.place用于设置值

In [1011]: w=np.empty_like(x)
In [1014]: np.place(w,w.mask,999)
In [1015]: np.place(w,~w.mask,[1,2,3])
In [1016]: w
Out[1016]: 
masked_array(data =
 [[-- -- 1]
 [2 3 --]],
             mask =
 [[ True  True False]
 [False False  True]],
       fill_value = 999999)
In [1017]: w.data
Out[1017]: 
array([[999, 999,   1],
       [  2,   3, 999]])

=====================

看着

https://github.com/numpy/numpy/blob/master/numpy/ma/core.py 
class _MaskedBinaryOperation:

此类用于实现 masked ufunc。它评估ufuncat 有效单元格（非屏蔽）并返回一个带有有效单元格的新屏蔽数组，使屏蔽值保持不变（从原始）

例如使用简单的掩码数组，+1不会更改掩码值。

In [1109]: z=np.ma.masked_equal([1,0,2],0)
In [1110]: z
Out[1110]: 
masked_array(data = [1 -- 2],
             mask = [False  True False],
       fill_value = 0)
In [1111]: z.data
Out[1111]: array([1, 0, 2])
In [1112]: z+1
Out[1112]: 
masked_array(data = [2 -- 3],
             mask = [False  True False],
       fill_value = 0)
In [1113]: _.data
Out[1113]: array([2, 0, 3])
In [1114]: z.compressed()+1
Out[1114]: array([2, 3])

_MaskedUnaryOperation可能更容易理解，因为它只需要使用 1 个掩码数组。

例如，常规日志的掩码 0 值存在问题：

In [1115]: z.log()
...
/usr/local/bin/ipython3:1: RuntimeWarning: divide by zero encountered in log
  #!/usr/bin/python3
Out[1116]: 
masked_array(data = [0.0 -- 0.6931471805599453],
             mask = [False  True False],
       fill_value = 0)

但是被屏蔽的日志会跳过被屏蔽的条目：

In [1117]: np.ma.log(z)
Out[1117]: 
masked_array(data = [0.0 -- 0.6931471805599453],
             mask = [False  True False],
       fill_value = 0)
In [1118]: _.data
Out[1118]: array([ 0.        ,  0.        ,  0.69314718])

哎呀——_MaskedUnaryOperation可能没那么有用。它评估ufuncat all values np.ma.getdata(z)，并带有errstate阻止警告的上下文。然后它使用掩码将掩码值复制到结果 ( np.copyto(result, d, where=m))。

于 2016-08-09T16:57:39.547 回答