带有rgb颜色的3D体素/体积图

演示使用ax.voxels可视化颜色空间的各个部分

带有rgb颜色的3D体素/体积图示例

  1. import matplotlib.pyplot as plt
  2. import numpy as np
  4. # This import registers the 3D projection, but is otherwise unused.
  5. from mpl_toolkits.mplot3d import Axes3D  # noqa: F401 unused import
  8. def midpoints(x):
  9.     sl = ()
  10.     for i in range(x.ndim):
  11.         x = (x[sl + np.index_exp[:-1]] + x[sl + np.index_exp[1:]]) / 2.0
  12.         sl += np.index_exp[:]
  13.     return x
  15. # prepare some coordinates, and attach rgb values to each
  16. r, g, b = np.indices((17, 17, 17)) / 16.0
  17. rc = midpoints(r)
  18. gc = midpoints(g)
  19. bc = midpoints(b)
  21. # define a sphere about [0.5, 0.5, 0.5]
  22. sphere = (rc - 0.5)**2 + (gc - 0.5)**2 + (bc - 0.5)**2 < 0.5**2
  24. # combine the color components
  25. colors = np.zeros(sphere.shape + (3,))
  26. colors[..., 0] = rc
  27. colors[..., 1] = gc
  28. colors[..., 2] = bc
  30. # and plot everything
  31. fig = plt.figure()
  32. ax = fig.gca(projection='3d')
  33. ax.voxels(r, g, b, sphere,
  34.           facecolors=colors,
  35.           edgecolors=np.clip(2*colors - 0.5, 0, 1),  # brighter
  36.           linewidth=0.5)
  37. ax.set(xlabel='r', ylabel='g', zlabel='b')
  39. plt.show()

下载这个示例