importmatplotlib.pyplotaspltimportnumpyasnpimportmatplotlib.colorsdefmidpoints(x):sl=()foriinrange(x.ndim):x=(x[sl+np.index_exp[:-1]]+x[sl+np.index_exp[1:]])/2.0sl+=np.index_exp[:]returnx# prepare some coordinates, and attach rgb values to eachr,theta,z=np.mgrid[0:1:11j,0:np.pi*2:25j,-0.5:0.5:11j]x=r*np.cos(theta)y=r*np.sin(theta)rc,thetac,zc=midpoints(r),midpoints(theta),midpoints(z)# define a wobbly torus about [0.7, *, 0]sphere=(rc-0.7)**2+(zc+0.2*np.cos(thetac*2))**2<0.2**2# combine the color componentshsv=np.zeros(sphere.shape+(3,))hsv[...,0]=thetac/(np.pi*2)hsv[...,1]=rchsv[...,2]=zc+0.5colors=matplotlib.colors.hsv_to_rgb(hsv)# and plot everythingax=plt.figure().add_subplot(projection='3d')ax.voxels(x,y,z,sphere,facecolors=colors,edgecolors=np.clip(2*colors-0.5,0,1),# brighterlinewidth=0.5)plt.show()

DownloadJupyternotebook:voxels_torus.ipynb

DownloadPythonsourcecode:voxels_torus.py

Downloadzipped:voxels_torus.zip