python堆叠条形图_在matplotlib中创建堆叠的圆柱条形图

from__future__importprint_functionfrom__future__importdivisionfrom__future__importabsolute_importfrommpl_toolkits.mplot3dimportAxes3Dimportnumpyimportmatplotlibimportmatplotlib.pyplotaspltdefplot_cylinder_element(x,z,dz,rx=5,ry=5,color="b"):"""

x: left, right

z: start height

dz: height of cylinder

rx, ry = radius of width (x) and depth (y)

color = color

Inspired by:

http://matplotlib.1069221.n5.nabble.com/plot-surface-shading-and-clipping-error-td14031.html

"""N=100# number of elements# a lower stride will give more faces. A cylinder with 4 faces is a cube :)# I think with N=100 and rstride=2, it will have 50 faces# cstride is the height, rstride the circlecstride_side=1000# only 1 element neededrstride_side=1# many elements to make a nice cylinder shapecstride_top=10rstride_top=10# parameters of cylinderphi=numpy.linspace(0,2*numpy.pi,N)_r=numpy.ones(N)_h=numpy.linspace(0,1,N)# cylinder_x=rx*numpy.outer(numpy.cos(phi),_r)+x

_y=ry*numpy.outer(numpy.sin(phi),_r)_z=dz*numpy.outer(numpy.ones(numpy.size(_r)),_h)+z

ax.plot_surface(_x,_y,_z,rstride=rstride_side,cstride=cstride_side,linewidth=0,alpha=1,color=color)# to cover the gaps between the faces, plot the cylinder again at a slightly smaller radius_x*=0.99_y*=0.99ax.plot_surface(_x,_y,_z,rstride=rstride_side+1,cstride=cstride_side+1,linewidth=0,alpha=1,color=color)# top_x=rx*numpy.outer(numpy.cos(phi),_h)+x

_y=ry*numpy.outer(numpy.sin(phi),_h)_z=numpy.zeros([N,N])+z+dz+0.1ax.plot_surface(_x,_y,_z,rstride=rstride_top,cstride=cstride_top,linewidth=0,alpha=1,color=color)# plot again with different stride to mask the gapsax.plot_surface(_x,_y,_z,rstride=rstride_side+1,cstride=cstride_side+1,linewidth=0,alpha=1,color=color)defplot_cylinder(x,z,rx=5,ry=5):"""

x: left-right for each cylinder

z: list height difference (ie. not cumulative)

"""# list with colorscolors=["b","g","r","c","y","k"]# plot cylinder elements_z=0foriinrange(len(z)):plot_cylinder_element(x,_z,z[i],rx=rx,ry=ry,color=colors[i%len(colors)])_z+=z[i]defcylinder_plot(z,r=10,dr=30):"""

z: list of different cylinders with for each a list height difference (ie. not cumulative)

r: radius

dr: distance between cylinders

"""# different cylinders next to each otherx=numpy.arange(len(z))*dr# possible difference between width (x) and depth (y)rx=r

ry=r# make cylindersforiinrange(len(z)):plot_cylinder(x[i],z[i],rx=rx,ry=ry)# close earlier plotsplt.close("all")# make figurefig=plt.figure()ax=Axes3D(fig)# set 3D-viewax.view_init(elev=10,azim=280)# make 3 cylinders, with a different number of elementscylinder_plot([[5,10,5],[3,5],[1,2,3,4]])# set the labelsax.set_xlabel('X')ax.set_ylabel('Y')ax.set_zlabel('Z')# showplt.show()