python绘制函数x^2+y^2的3D图像

meshgrid()函数：
初始散点数据处理成xy网格数据

plot_surface(x, y, f(x, y)，rstride=1, cstride=1, cmap=‘rainbow’)
x、y数据决定坐标点，f(x,y)数据决定 x、y 坐标点对应的高度；rstride（row）指定行的跨度；cstride(column)指定列的跨度；cmap 设置颜色。

程序代码：

from matplotlib import pyplot as plt
import numpy as np
# 导入可以创建三维坐标的库
from mpl_toolkits.mplot3d import Axes3D
# 设置字体，使可以输入中文
plt.rcParams['font.sans-serif']=['STSong']
# 绘制三维坐标
# 创建一个新的matplotlib.figure.Figure并为其添加一个类型为Axes3D的新轴
fig = plt.figure()
ax = Axes3D(fig)