1. import matplotlib.pyplot as plt
    2. import numpy as np
    3. from mpl_toolkits.mplot3d import Axes3D
    5. # prepare some coordinates
    6. x, y, z = np.indices((16, 16, 16))
    8. # draw cuboids in the top left and bottom right corners, and a link between them
    9. #cube1 =  (x < 230)& (x > 0) &  (y < 200) & (y > 0)  & (z < 170) & (z > 0)
    10. cube1 = (x > 3) & (x < 12) & (y > 4) & (y < 13) & (z > 5) & (z < 14) 
    11. #cube2 = (x >= 5) & (y >= 5) & (z >= 5)
    12. #link = abs(x - y) + abs(y - z) + abs(z - x) <= 2
    14. # combine the objects into a single boolean array
    15. voxels = cube1 # | cube2 | link
    17. # set the colors of each object
    18. colors = np.empty(voxels.shape, dtype=object)
    19. #colors[link] = 'red'
    20. colors[cube1] = 1#"gray"#'blue'
    21. #colors[cube2] = 'green'
    23. colors = np.random.random((16, 16, 16, 3))
    25. # and plot everything
    26. fig = plt.figure(dpi=800)
    28. ax = Axes3D(fig)    # 一个轴对象
    29. ax.grid()
    30. ax.text(-1, -1, -1, "(0, 0, 0)")
    31. ax.text(15, 15, 15, "(256, 256, 256)")
    32. ax.text(17, -1, -1, "(256, 0, 0)")
    35. ax.set_xlabel('X axis')
    36. ax.set_ylabel('Y axis')
    37. ax.set_zlabel('Z axis')
    39. ax.set_xticks([])
    40. ax.set_yticks([])
    41. ax.set_zticks([])    # 轴对象的设置标签方法
    42. ax.voxels(voxels, facecolors=colors, edgecolor='k')
    44. plt.savefig("3d.jpg")
    45. plt.show()

    绘制三维图 - 图1