python根据指定的数据(实验表格)绘制三维曲面图

本文记录对于给定数据用python绘制三维曲面图的一个可用代码

一、需求

• 将上述表格中的数据绘制成三维曲面图

二、代码及注意点

• 首先输入X轴及Y轴数据

X = np.array([3,4,5,6,7,8,9,10])
Y = np.array([1596,1196,798,624,416,208])
X, Y = np.meshgrid(X, Y)
print("网格化后的X=",X)
print("X维度信息",X.shape)
print("网格化后的Y=",Y)
print("Y维度信息", Y.shape)

• 先要将X,Y变成维度相同的二维数据
• 输入Z轴的数据，也就是我们需要绘制的数据,并将其调整为与XY网格同样的维度

Z = np.array(
    [
    [1.85,	2.47,	3.64	,5.26,	7.03,	14.32],
    [3.09,	4.13,	6.19	,7.81,	11.81,	24.04],
    [4.67,	6.24,	9.24,	11.95	,17.99,	36.64],
    [6.59	,8.84	,13.03,	16.73,	24.98,	51.87],
    [8.98	,11.82,	17.41	,22.39	,33.79	,68.52],
    [11.47	,15.47	,22.51	,28.71,	43.33	,88.21],
    [14.34,	19.38,	28.09,	36.19	,54.22,	112.77],
    [17.95,	23.76	,34.76	,44.23,	66.29	,136.09]
]
)
print("维度调整前的Z轴数据维度",Z.shape)
Z = Z.T
print("维度调整后的Z轴数据维度",Z.shape)

• 最后，开始绘制

# 绘制三维曲面图
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')
#设置三个坐标轴信息
ax.set_xlabel('Dimiension', color='b')
ax.set_ylabel('Frequency (MHz)', color='g')
ax.set_zlabel('Time (ms)', color='r')

plt.draw()
plt.show()
plt.savefig('3D.jpg')

• 效果，注意这个图像是可以随意拖动旋转的，可以调整到较好的效果，然后保存为指定格式
  

三、完整代码

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

fig = plt.figure()
ax = Axes3D(fig)

# 数据录入
X = np.array([3,4,5,6,7,8,9,10])
Y = np.array([1596,1196,798,624,416,208])
X, Y = np.meshgrid(X, Y)
print("网格化后的X=",X)
print("X维度信息",X.shape)
print("网格化后的Y=",Y)
print("Y维度信息", Y.shape)

Z = np.array(
    [
    [1.85,	2.47,	3.64	,5.26,	7.03,	14.32],
    [3.09,	4.13,	6.19	,7.81,	11.81,	24.04],
    [4.67,	6.24,	9.24,	11.95	,17.99,	36.64],
    [6.59	,8.84	,13.03,	16.73,	24.98,	51.87],
    [8.98	,11.82,	17.41	,22.39	,33.79	,68.52],
    [11.47	,15.47	,22.51	,28.71,	43.33	,88.21],
    [14.34,	19.38,	28.09,	36.19	,54.22,	112.77],
    [17.95,	23.76	,34.76	,44.23,	66.29	,136.09]
]
)
print("维度调整前的Z轴数据维度",Z.shape)
Z = Z.T
print("维度调整后的Z轴数据维度",Z.shape)

# 绘制三维曲面图
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')
#设置三个坐标轴信息
ax.set_xlabel('Dimiension', color='b')
ax.set_ylabel('Frequency (MHz)', color='g')
ax.set_zlabel('Time (ms)', color='r')

plt.draw()
plt.show()
plt.savefig('3D.jpg')