欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

如何用Python绘制3D柱形图

程序员文章站 2022-07-02 20:18:10
本文主要讲解如何使用python绘制三维的柱形图,如下图源代码如下:import numpy as npimport matplotlib.pyplot as pltfrom mpl_toolkits...

本文主要讲解如何使用python绘制三维的柱形图,如下图

如何用Python绘制3D柱形图

源代码如下:

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

#构造需要显示的值
x=np.arange(0, 5, step=1)#x轴的坐标
y=np.arange(0, 9, step=1)#y轴的坐标
#设置每一个(x,y)坐标所对应的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)#网格化坐标
x, y=xx.ravel(), yy.ravel()#矩阵扁平化
bottom=np.zeros_like(x)#设置柱状图的底端位值
z=z.ravel()#扁平化矩阵

width=height=1#每一个柱子的长和宽

#绘图设置
fig=plt.figure()
ax=fig.gca(projection='3d')#三维坐标轴
ax.bar3d(x, y, bottom, width, height, z, shade=true)#
#坐标轴设置
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z(value)')
plt.show()

代码解读:

1、构造需要显示的数据

如下图所示,x坐标取值为[0,1,2,3,4],y坐标取值为[0,1,2,3,4,5,6,7,8],每一个(x,y)组合的值z=x+y,所需要绘制的图就是在x,y所对应的坐标位置上面根据z的值来绘制柱形图。

如何用Python绘制3D柱形图

2、坐标设置

将坐标网格化, x=[0,1,2,3,4],y=[0,1,2,3,4,5,6,7,8]网格化的结果,如下图所示。可以理解为,x参照y进行了广播(broadcast), y参照x进行了广播,分别添加了一个新的维度,并且进行复制。结合xx和yy就得到了所有的x,y的坐标组合,经过扁平化之后这种对应关系保持不变。代码中的bottom是用来设置绘制每一个柱子的z坐标的,在此全部都设置为0,而width和height设置每一个柱子的长和宽。

如何用Python绘制3D柱形图

3、bar3d

ax.bar3d(x, y, bottom, width, height, z, shade=true):
x,y:柱子在xy平面的起始坐标
bottom:柱子在z轴上的起始坐标
width、height:柱子的长宽
z:数字沿z轴的长度
shade:是否显示阴影(设置为true立体效果会更好)

shde=true

如何用Python绘制3D柱形图

shade=false

如何用Python绘制3D柱形图

如果需要绘制多个子图,并且里面每一个子图都是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柱形图的详细内容,更多关于python绘制柱形图的资料请关注其它相关文章!