vispy：与颜色相同的两个数据集

Question

我有两个3d numpy数组，我试图通过一个可见的三维散射以两种不同的颜色绘制它们。

我已经熟悉了如何通过分散在vispy上设置数据：

scatter.set_data（数据）

其中数据采用numpy数组的形式。但是，我需要绘制两种不同颜色的两种不同数据集，我觉得这里有一些非常明显的东西。将两个点集视为一个数据集我很好，但是如何为点组获得两种不同的颜色？

Answer 1

我修改了vispy repo中的分散示例以执行您要求的操作。

结果：

代码：

# pylint: disable=no-member
""" scatter using MarkersVisual """

import numpy as np
import sys

from vispy import app, visuals, scene


# build your visuals, that's all
Scatter3D = scene.visuals.create_visual_node(visuals.MarkersVisual)

# The real-things : plot using scene
# build canvas
canvas = scene.SceneCanvas(keys='interactive', show=True)

# Add a ViewBox to let the user zoom/rotate
view = canvas.central_widget.add_view()
view.camera = 'turntable'
view.camera.fov = 45
view.camera.distance = 500

# data
n = 500
# generate 2 point clouds
cloud1 = np.random.rand(n, 3) * 100
cloud2 = np.random.rand(n, 3) * 100
# cloud1 -> orange
# cloud2 -> white
color1 = np.array([[1, 0.4, 0]] * n)
color2 = np.ones((n, 3))

# stack point clouds and colors
pos = np.vstack((cloud1, cloud2))
colors = np.vstack((color1, color2))


# plot ! note the parent parameter
p1 = Scatter3D(parent=view.scene)
p1.set_gl_state('translucent', blend=True, depth_test=True)
p1.set_data(pos, face_color=colors, symbol='o', size=10,
            edge_width=0.5, edge_color='blue')

# run
if __name__ == '__main__':
    if sys.flags.interactive != 1:
        app.run()