如何用Python繪制3D柱形圖
本文主要講解如何使用python繪制三維的柱形圖,如下圖
源代碼如下:
import numpy as npimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3D#構(gòu)造需要顯示的值X=np.arange(0, 5, step=1)#X軸的坐標(biāo)Y=np.arange(0, 9, step=1)#Y軸的坐標(biāo)#設(shè)置每一個(X,Y)坐標(biāo)所對應(yīng)的Z軸的值,在這邊Z(X,Y)=X+YZ=np.zeros(shape=(5, 9))for i in range(5): for j in range(9): Z[i, j]=i+jxx, yy=np.meshgrid(X, Y)#網(wǎng)格化坐標(biāo)X, Y=xx.ravel(), yy.ravel()#矩陣扁平化bottom=np.zeros_like(X)#設(shè)置柱狀圖的底端位值Z=Z.ravel()#扁平化矩陣width=height=1#每一個柱子的長和寬#繪圖設(shè)置fig=plt.figure()ax=fig.gca(projection=’3d’)#三維坐標(biāo)軸ax.bar3d(X, Y, bottom, width, height, Z, shade=True)##坐標(biāo)軸設(shè)置ax.set_xlabel(’X’)ax.set_ylabel(’Y’)ax.set_zlabel(’Z(value)’)plt.show()
代碼解讀:
1、構(gòu)造需要顯示的數(shù)據(jù)
如下圖所示,X坐標(biāo)取值為[0,1,2,3,4],Y坐標(biāo)取值為[0,1,2,3,4,5,6,7,8],每一個(X,Y)組合的值Z=X+Y,所需要繪制的圖就是在X,Y所對應(yīng)的坐標(biāo)位置上面根據(jù)Z的值來繪制柱形圖。
2、坐標(biāo)設(shè)置
將坐標(biāo)網(wǎng)格化, X=[0,1,2,3,4],Y=[0,1,2,3,4,5,6,7,8]網(wǎng)格化的結(jié)果,如下圖所示。可以理解為,X參照Y進(jìn)行了廣播(broadcast), Y參照X進(jìn)行了廣播,分別添加了一個新的維度,并且進(jìn)行復(fù)制。結(jié)合xx和yy就得到了所有的X,Y的坐標(biāo)組合,經(jīng)過扁平化之后這種對應(yīng)關(guān)系保持不變。代碼中的bottom是用來設(shè)置繪制每一個柱子的Z坐標(biāo)的,在此全部都設(shè)置為0,而width和height設(shè)置每一個柱子的長和寬。
3、bar3d
ax.bar3d(X, Y, bottom, width, height, Z, shade=True):X,Y:柱子在XY平面的起始坐標(biāo)bottom:柱子在Z軸上的起始坐標(biāo)width、height:柱子的長寬Z:數(shù)字沿Z軸的長度shade:是否顯示陰影(設(shè)置為True立體效果會更好)
shde=True
shade=False
如果需要繪制多個子圖,并且里面每一個子圖都是3D的,只需要將上述代碼進(jìn)行些許修改即可,案例如下:
fig=plt.figure() ax=fig.add_subplot(1, 3, 1, projection=’3d’) ax.bar3d(x, y, z, width, height, top, shade=True) ax=fig.add_subplot(1, 3, 2, projection=’3d’) ax.bar3d(x, y, z, width, height, top2, shade=True) ax=fig.add_subplot(1, 3, 3, projection=’3d’) ax.bar3d(x, y, z, width, height, top3, shade=True) plt.show()
以上就是如何用Python繪制3D柱形圖的詳細(xì)內(nèi)容,更多關(guān)于python繪制柱形圖的資料請關(guān)注好吧啦網(wǎng)其它相關(guān)文章!
相關(guān)文章:
