2020-08-06 pythong根据实验数据绘制曲面

code

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')