python绘图基础-2

使用python绘制3维图
1、导入相应的包

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

2.设置变量

fig = plt.figure()
X = np.linspace(-4, 4, 100)
Y = np.linspace(-4, 4, 100)
X, Y = np.meshgrid(X, Y)
Z = X**2+Y**2

3绘图
3.1绘制曲面图

#绘制曲面图
ax = Axes3D(fig)
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')
plt.show()

3.2绘制等高线

#绘制等高线
bx = fig.add_subplot(111)
C = plt.contour(X, Y, Z)
#对等高线值进行标注
bx.clabel(C, alpha=1, cmap='rainbow',inline=1, fontsize=10)
plt.show()