使用Mayavi Python将颜色分配给points3d中的每个点

Question

我在3D空间中有一些要点。每个点都有一种颜色，该颜色是使用以下公式计算的-

// pack r/g/b into rgb
uint8_t r = 255, g = 0, b = 0;    // Example: Red color
uint32_t rgb = ((uint32_t)r << 16 | (uint32_t)g << 8 | (uint32_t)b);

如上所示，RGB颜色被打包为一个值。我正在尝试使用mayavi python可视化这些点。请参见下面的代码段-

from mayavi.mlab import *
import numpy as np

N = 10
# generate random points and colors (just for debugging)
(x, y, z) = np.random.random((3, N))
colors = np.random.random(N)

nodes = points3d(x, y, z, scale_factor=0.1)
nodes.glyph.scale_mode = 'scale_by_vector'
nodes.mlab_source.dataset.point_data.scalars = colors

show()

上面的代码使用随机颜色，并且显示以下输出-

但是，我想指定颜色而不是使用随机值。请注意，每个点都有颜色。在本文中，为了简化操作，我将使用以下功能生成相同颜色的点-

def pack_rgb(r, g, b):
    rgb = (r<<16) + (g<<8) + b
    return rgb

colors = [pack_rgb(0, 255, 0) for _ in range(N)]

这将生成红色的点，而不是绿色的点，如下所示-

这是怎么回事？ 我的目标是在mayavi python中可视化着色的点，其中每个点都有RGB颜色。

Answer 1

使用Mayavi可以轻松地可视化3D空间中的点。可以分别为每个点分配RGB颜色。我们还可以为每个点设置比例因子。我从Mayavi社区找到了以下解决方案，我想在这里分享它-

from mayavi import mlab
import numpy as np

n = 100 # number of points
x, y, z = np.random.random((3, n))
rgba = np.random.randint(0, 256, size=(n, 4), dtype=np.uint8)
rgba[:, -1] = 255 # no transparency
pts = mlab.pipeline.scalar_scatter(x, y, z) # plot the points
pts.add_attribute(rgba, 'colors') # assign the colors to each point
pts.data.point_data.set_active_scalars('colors')
g = mlab.pipeline.glyph(pts)
g.glyph.glyph.scale_factor = 0.1 # set scaling for all the points
g.glyph.scale_mode = 'data_scaling_off' # make all the points same size

mlab.show()

请注意，以上代码需要Mayavi 4.6.1或更高版本。