如何在Mayavi的网格图中使用掩码？

Question

如何在Mayavi的网格图中使用掩码？

4

我正在使用Python和Mayavi2进行3D绘图。我正在使用mesh命令绘制一个球体，现在我想将球体的某些面板涂成不同的颜色。似乎mask选项就是为此而设的，但我无法使其正常工作（整个球体都被重新涂色）。

参考网页：http://docs.enthought.com/mayavi/mayavi/auto/mlab_helper_functions.html#mayavi.mlab.mesh “mask:用于抑制一些数据点的布尔掩码数组”

我的代码如下：

mesh(X,Y,Z, color = (1,1,1), opacity=0.5)

将整个球体涂成白色，然后

mesh(X,Y,Z, color = (1,0,1), mask = active_region, opacity=0.5)

把一些部分涂成紫色，其中包括:

active_region = [[False False False False False False  True]
                 [False False False False False False  True]
                 [False False False False False  True  True]
                 [False False False  True  True  True  True]
                 [False False  True  True  True  True  True]
                 [False False False  True  True  True  True]
                 [False False False False False False  True]]

但是这会导致完全紫色的球体。X、Y和Z都是形状为（7,7）的数组，就像active_region一样。我做错了什么？

- capybaralet

嗯，应该可以了。如果所有的值都是True或False会发生什么？ - M4rtini

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Prabhu Ramachandran · Accepted Answer

掩码基于标量而不是设置了纯色时运作。请尝试使用以下方法代替：

import numpy as np
m = mesh(X,Y,Z, mask=active_region, opacity=0.5)
m.mlab_source = ones_like(X)

掩蔽功能通过在掩蔽的位置设置标量为NaN来实现。绘制的几何形状相同，但如果标量为NaN，则不会显示掩蔽区域。

无论如何，这里是一个完整的工作示例供您参考：

import numpy as np
phi, theta = np.mgrid[0:np.pi:100j, 0:2 * np.pi:100j]
x = np.sin(phi) * np.cos(theta)
y = np.sin(phi) * np.sin(theta)
z = np.cos(phi)
mask = np.zeros_like(x).astype(bool)
mask[::5] = True
mask[:,::5] = True

from mayavi import mlab
mlab.mesh(x, y, z, scalars=z, mask=mask)

请注意，您不需要提供标量值，只是如果提供了显式颜色，就不会进行掩码操作。此处的标量值仅设置为z，您可以使用任何兼容的数组，甚至是通过ones_like(X)设置的常量值。

事实证明，如果您有一个恒定的标量值，VTK 5.6似乎无法正确地屏蔽您的点；如果您有VTK 5.10，则可以执行以下操作：

m = mlab.mesh(x, y, z, scalars=scalars, mask=mask)
m.module_manager.scalar_lut_manager.lut.nan_color = 0, 0, 0, 0

否则，请使用scalar=z并设置一个恒定颜色查找表。您可以通过在管道编辑器上单击“颜色和图例”，然后单击“启动lut编辑器”来创建自己的颜色映射表。要使用此自定义颜色映射，请执行以下操作：

scalars = np.ones_like(x)
m = mlab.mesh(x, y, z, scalars=scalars, mask=mask)
m.module_manager.scalar_lut_manager.lut_mode = 'file'
m.module_manager.scalar_lut_manager.file_name = '/path/to/your.lut'

如果你不想与LUT编辑器及其它繁琐的东西打交道，这里有一个纯粹的Numpy版本，您可以直接从Python中使用。

m = mlab.mesh(x, y, z, scalars=z, mask=mask)
# Lets make some colors:
#    this is an array of (R, G, B, A) values (each in range 0-255), there should
#    be at least 2 colors in the array.
colors = np.zeros((2, 4), dtype='uint8')
# Set the green value to constant.
colors[:,1] = 255
# the alpha value to fully opaque.
colors[:,3] = 255
# Now use this colormap.
m.module_manager.scalar_lut_manager.lut.table = colors

请注意，颜色数组应该至少有2个颜色，可以有任意多个。对于一个固定的颜色，两个就足够了。

希望这有帮助。