具有圆柱坐标的3D体素/体积图

演示使用ax.voxels的x，y，z参数。

具有圆柱坐标的3D体素/体积图示例

  1. import matplotlib.pyplot as plt
  2. import matplotlib.colors
  3. import numpy as np
  5. # This import registers the 3D projection, but is otherwise unused.
  6. from mpl_toolkits.mplot3d import Axes3D  # noqa: F401 unused import
  9. def midpoints(x):
  10.     sl = ()
  11.     for i in range(x.ndim):
  12.         x = (x[sl + np.index_exp[:-1]] + x[sl + np.index_exp[1:]]) / 2.0
  13.         sl += np.index_exp[:]
  14.     return x
  16. # prepare some coordinates, and attach rgb values to each
  17. r, theta, z = np.mgrid[0:1:11j, 0:np.pi*2:25j, -0.5:0.5:11j]
  18. x = r*np.cos(theta)
  19. y = r*np.sin(theta)
  21. rc, thetac, zc = midpoints(r), midpoints(theta), midpoints(z)
  23. # define a wobbly torus about [0.7, *, 0]
  24. sphere = (rc - 0.7)**2 + (zc + 0.2*np.cos(thetac*2))**2 < 0.2**2
  26. # combine the color components
  27. hsv = np.zeros(sphere.shape + (3,))
  28. hsv[..., 0] = thetac / (np.pi*2)
  29. hsv[..., 1] = rc
  30. hsv[..., 2] = zc + 0.5
  31. colors = matplotlib.colors.hsv_to_rgb(hsv)
  33. # and plot everything
  34. fig = plt.figure()
  35. ax = fig.gca(projection='3d')
  36. ax.voxels(x, y, z, sphere,
  37.           facecolors=colors,
  38.           edgecolors=np.clip(2*colors - 0.5, 0, 1),  # brighter
  39.           linewidth=0.5)
  41. plt.show()

下载这个示例