Wie colormap in VisPy verwendet surfaceplot

Question

Ich VisPy mit einer Oberfläche zu zeichnen, aber die Funktion scene.visuals.SurfacePlot hat nur den Parameter color, ohne den Parameter colormap. Kann jemand colormap im Oberflächenplotten verwenden?

Answer 1

Momentan scheint es einen bug zu geben, der den colors Parameter von SurfacePlot nicht funktioniert. Wenn man der Oberfläche alles andere als eine feste Farbe geben möchte, müsste man wahrscheinlich diesen Weg gehen. Es ist also zu diesem Zeitpunkt möglicherweise nicht möglich, den Quellcode zu ändern.

Für das, was es wert ist, glaube ich, die folgenden, wenn dieser Fehler nicht vorhanden war funktionieren würde:

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) 

Grundsätzlich Sie die Daten normalisieren ein zu zwischen 0 und 1, und ordnen dann diese Werte Farbkarte. Dadurch wird ein 3D-Array mit RGB-Farben zurückgegeben, die Sie an den Parameter colors von SurfacePlot übergeben können.

Answer 2

Die Farben müssen auf einer Vertexbasis gesetzt werden, dh Vertex_colors. Eine Möglichkeit, dies zu erreichen, ist wie folgt;

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() 

Beachten Sie die Verwendung des expliziten Setter set_vertex_colors hat gut funktioniert. Leider (möglicherweise ein Fehler) beim Übergeben c an surface(z, vertex_colors=c) lief ohne Fehler, aber änderte nicht die pro Vertex Farben. Fazit: Verwenden Sie explizit set_vertex_colors.