三维体素绘制Numpy的Logo

演示使用坐标不均匀的ax.voxels

三维体素绘制Numpy的Logo示例

  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 explode(data):
  9.     size = np.array(data.shape)*2
  10.     data_e = np.zeros(size - 1, dtype=data.dtype)
  11.     data_e[::2, ::2, ::2] = data
  12.     return data_e
  14. # build up the numpy logo
  15. n_voxels = np.zeros((4, 3, 4), dtype=bool)
  16. n_voxels[0, 0, :] = True
  17. n_voxels[-1, 0, :] = True
  18. n_voxels[1, 0, 2] = True
  19. n_voxels[2, 0, 1] = True
  20. facecolors = np.where(n_voxels, '#FFD65DC0', '#7A88CCC0')
  21. edgecolors = np.where(n_voxels, '#BFAB6E', '#7D84A6')
  22. filled = np.ones(n_voxels.shape)
  24. # upscale the above voxel image, leaving gaps
  25. filled_2 = explode(filled)
  26. fcolors_2 = explode(facecolors)
  27. ecolors_2 = explode(edgecolors)
  29. # Shrink the gaps
  30. x, y, z = np.indices(np.array(filled_2.shape) + 1).astype(float) // 2
  31. x[0::2, :, :] += 0.05
  32. y[:, 0::2, :] += 0.05
  33. z[:, :, 0::2] += 0.05
  34. x[1::2, :, :] += 0.95
  35. y[:, 1::2, :] += 0.95
  36. z[:, :, 1::2] += 0.95
  38. fig = plt.figure()
  39. ax = fig.gca(projection='3d')
  40. ax.voxels(x, y, z, filled_2, facecolors=fcolors_2, edgecolors=ecolors_2)
  42. plt.show()

下载这个示例