如何用Python繪制3D柱形圖

更新時間：2020年09月16日 14:33:39 作者：NewRookie

這篇文章主要介紹了如何用Python繪制3D柱形圖，幫助大家更好的利用python實現(xiàn)數(shù)據(jù)可視化，感興趣的朋友可以了解下

本文主要講解如何使用python繪制三維的柱形圖，如下圖

源代碼如下：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

#構(gòu)造需要顯示的值
X=np.arange(0, 5, step=1)#X軸的坐標
Y=np.arange(0, 9, step=1)#Y軸的坐標
#設(shè)置每一個（X，Y）坐標所對應(yīng)的Z軸的值，在這邊Z（X，Y）=X+Y
Z=np.zeros(shape=(5, 9))
for i in range(5):
  for j in range(9):
    Z[i, j]=i+j

xx, yy=np.meshgrid(X, Y)#網(wǎng)格化坐標
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')#三維坐標軸
ax.bar3d(X, Y, bottom, width, height, Z, shade=True)#
#坐標軸設(shè)置
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z(value)')
plt.show()

代碼解讀：

1、構(gòu)造需要顯示的數(shù)據(jù)

如下圖所示，X坐標取值為[0,1,2,3,4]，Y坐標取值為[0,1,2,3,4,5,6,7,8]，每一個(X,Y)組合的值Z=X+Y，所需要繪制的圖就是在X，Y所對應(yīng)的坐標位置上面根據(jù)Z的值來繪制柱形圖。

2、坐標設(shè)置

將坐標網(wǎng)格化, X=[0,1,2,3,4]，Y=[0,1,2,3,4,5,6,7,8]網(wǎng)格化的結(jié)果，如下圖所示?？梢岳斫鉃?，X參照Y進行了廣播（broadcast）, Y參照X進行了廣播，分別添加了一個新的維度，并且進行復制。結(jié)合xx和yy就得到了所有的X，Y的坐標組合，經(jīng)過扁平化之后這種對應(yīng)關(guān)系保持不變。代碼中的bottom是用來設(shè)置繪制每一個柱子的Z坐標的，在此全部都設(shè)置為0，而width和height設(shè)置每一個柱子的長和寬。

3、bar3d

ax.bar3d(X, Y, bottom, width, height, Z, shade=True)：
X，Y：柱子在XY平面的起始坐標
bottom:柱子在Z軸上的起始坐標
width、height:柱子的長寬
Z：數(shù)字沿Z軸的長度
shade:是否顯示陰影（設(shè)置為True立體效果會更好）

shde=True

shade=False

如果需要繪制多個子圖，并且里面每一個子圖都是3D的，只需要將上述代碼進行些許修改即可，案例如下：

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柱形圖的詳細內(nèi)容，更多關(guān)于python繪制柱形圖的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: