如何在VisPy surfaceplot中使用colormap

Question

我正在使用VisPy绘制曲面，但函数scene.visuals.SurfacePlot只有参数color，没有参数colormap。有没有人知道如何在表面绘图中使用colormap？

Answer 1

目前，看起来有bug这是保持SurfacePlot的colors参数不起作用。除了纯色以外，任何表面都可能不得不走上这条路。因此，目前可能无法修改源代码。

对于它的价值，我相信如果这个错误是不存在以下将工作：

import vispy.plot as vp 
from vispy import color 

fig = vp.Fig(show=False) 
cnorm = z/abs(np.amax(z)) 
colors = color.get_colormap("hsl").map(cnorm).reshape(z.shape + (-1,)) 
fig[0, 0].surface(z, x=x, y=y, colors=colors) 

基本上，数据标准化为0和1之间，然后映射这些值到色彩表。这将返回一个RGB颜色的三维数组，您可以将其传递给SurfacePlot的colors参数。

Answer 2

必须在每个顶点基础上设置颜色，即vertex_colors。一种可以实现的方式如下：

import vispy.plot as vp 
from vispy import color 
from vispy.util.filter import gaussian_filter 
import numpy as np 

z = np.random.normal(size=(250, 250), scale=200) 
z[100, 100] += 50000 
z = gaussian_filter(z, (10, 10)) 

fig = vp.Fig(show=False) 
cnorm = z/abs(np.amax(z)) 
c = color.get_colormap("hsl").map(cnorm).reshape(z.shape + (-1,)) 
c = c.flatten().tolist() 
c=list(map(lambda x,y,z,w:(x,y,z,w), c[0::4],c[1::4],c[2::4],c[3::4])) 

#p1 = fig[0, 0].surface(z, vertex_colors=c) # why doesn't vertex_colors=c work? 
p1 = fig[0, 0].surface(z) 
p1.mesh_data.set_vertex_colors(c) # but explicitly setting vertex colors does work? 

fig.show() 

注意使用显式设置set_vertex_colors工作正常。不幸的是（错误可能）通过c到surface(z, vertex_colors=c)运行没有错误，但没有改变每个顶点的颜色。结论：使用明确的set_vertex_colors。