mayavi在不改变大小的情况下设置colormap

mayavi_point3d.py

cl_rst = np.array(clustering.labels_)
print(frame_file)

fig = mayavi.mlab.figure(mlab.gcf(), bgcolor=(0, 0, 0), size=(640 * 2, 360 * 2))
nodes = mayavi.mlab.points3d(
    xx, yy, zz,
    # clustering.labels_,
    # mode="cube",
    mode="cube",
    # color=(0, 1, 0),
    colormap='spectral',
    figure=fig,
    scale_factor=1
)
# set color map in VTK
nodes.glyph.scale_mode = 'scale_by_vector'
nodes.mlab_source.dataset.point_data.scalars = cl_rst / max(cl_rst)