X, Y = np.meshgrid(X, Y) ax = Axes3D(fig) ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.vir

matplotlib三维绘图

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

X = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)    #将两个一维数组变成二维矩阵，将x转变为行向量，将y转为列向量
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)

fig = plt.figure()
ax = Axes3D(fig)   # 将图像转化为3D
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.viridis)    #rstride行跨度，cstride列跨度  cmap颜色映射表

plt.show()

import matplotlib.pyplot as plt
#导入3D包
from mpl_toolkits.mplot3d import Axes3D
#创建X、Y、Z坐标
X=[1,1,2,2]
Y=[3,4,4,3]
Z=[1,100,1,1]
fig = plt.figure()
#创建了一个A