在掩码阵列上操作，然后在同一阵列的不同部分上操作

Question

所以我试图在matplotlib中实现一个symlog样式的色阶，我这样做的方法是将我的值缩放到0-1区域，然后使用自定义colourbar刻度和标签。

我正在使用蒙版在三个部分中缩放我的值，但是它没有像我预期的那样工作;

我首先这样做：

negMask = ma.masked_greater_equal(Z, -10**thresh, copy=True).mask
posMask = ma.masked_less_equal(Z, 10**thresh, copy=True).mask
linMask = ma.masked_outside(Z, -10**thresh, 10**thresh, copy=True).mask

获得我想要的三个面具。然后我开始缩放到0-1区域：

sizeNeg = int(np.ceil(log10(-minZ)) - thresh)
sizePos = int(np.ceil(log10(maxZ)) - thresh)
sizeLin = 2
totSize = sizeNeg + sizePos + sizeLin

Z.mask = negMask
Z = (-np.log10(-Z) - thresh) / totSize

如果我停在这里并绘制它，那么它的行为与预期的一样，这里是情节的一小部分：

你可以看到它绘制负值，但不是正面部分 - 因为它们被掩盖了。

然后我将掩码更改为正值的掩码;

Z.mask = posMask
Z = (np.log10(Z) - thresh + sizeNeg + sizeLin) / totSize

但是当我这样做时，之前掩盖的所有值现在都只是1.0。那里的数据消失了。

这是我的意思的一个例子：

$ python
Python 2.6.6 (r266:84292, Sep 11 2012, 05:25:09) 
[GCC 4.4.6 20120305 (Red Hat 4.4.6-4)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import numpy as np
>>> from numpy.ma import masked_where as mw
>>> a = np.arange(0,10,1)
>>> a = mw(False, a)
>>> a
masked_array(data = [0 1 2 3 4 5 6 7 8 9],
             mask = False,
       fill_value = 999999)

>>> mask1 = mw(a>4,a).mask
>>> mask2 = mw(a<6,a).mask
>>> mask1
array([False, False, False, False, False,  True,  True,  True,  True,  True], dtype=bool)
>>> mask2
array([ True,  True,  True,  True,  True,  True, False, False, False, False], dtype=bool)
>>> a.mask = mask1
>>> a
masked_array(data = [0 1 2 3 4 -- -- -- -- --],
             mask = [False False False False False  True  True  True  True  True],
       fill_value = 999999)

>>> a = np.log10(a)
__main__:1: RuntimeWarning: divide by zero encountered in log10
>>> a
masked_array(data = [-- 0.0 0.301029995664 0.47712125472 0.602059991328 -- -- -- -- --],
             mask = [ True False False False False  True  True  True  True  True],
       fill_value = 999999)

>>> a.mask = mask2
>>> a
masked_array(data = [-- -- -- -- -- -- 1.0 1.0 1.0 1.0],
             mask = [ True  True  True  True  True  True False False False False],
       fill_value = 999999)

在屏蔽数组上使用np.log10会破坏屏蔽值，因此当您取消屏蔽它们时，它们只是1.0。

我知道我可以将我的阵列分成三个独立的阵列，每个区域一个，但之后如何将它们再次合并？