python - 使用 mlab.surf() 将蒙版与 Mayavi 3d 曲面图结合使用

Question

使用 mlab.surf() 函数绘制 3d 表面时，Nan 值的存在会导致奇怪的绘制行为。任何关于我可能做错的想法都将不胜感激。

In[37]: import mayavi
In[38]: print mayavi.__version__
4.4.0

from mayavi import mlab
j = np.copy('some 2d-array') # where j is shape (680, 680) and contains Nans
mlab.surf(j)

如果我用最小值替换 NaN，绘图看起来不错，但在我想要剪裁的边缘周围有一个人造平面:

j[np.isnan(j)]=j[~np.isnan(j)].min()
mlab.surf(j) # where j has been set to j.min() where j==np.nan

如果我然后添加一个对应于 j==j.min() 区域的 mask ，我会再次得到奇怪的输出(我也尝试过直接在重新定义 NaN 之前的数组):

m = np.ma.masked_where(j==j.min(), j).mask
mlab.surf(j, mask=m)

Answer 1

我会尝试这样的事情:

    #first mask all the NaN:
    maskedarray = np.ma.masked_invalid(j)
    from mayavi import mlab
    mlab.figure(1, fgcolor=(0, 0, 0), bgcolor=(1, 1, 1))        
    surH = mlab.surf(maskedarray, warp_scale='auto', mask=maskedarray.mask)    
    mlab.show()