下文给出了绘制函数  的 Python 代码。
很显然，这是一个三维图像。

【绘制三维图像的Python代码】

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
 
fig=plt.figure()
ax=Axes3D(fig)
x1=np.arange(-3.0, 3.0, 0.1)
x2=np.arange(-3.0, 3.0, 0.1)
X, Y=np.meshgrid(x1, x2)
Z=X**2+Y**2
plt.xlabel('x1')
plt.ylabel('x2')
ax.plot_surface(X, Y, Z, rstride=3, cstride=3, cmap='rainbow')
plt.show()

注意：
ax.plot_surface(X, Y, Z, rstride=3, cstride=3, cmap='rainbow') 用于绘制三维图像。其中：
rstride表示行之间的跨度，用来控制行平滑程度。
cstride表示列之间的跨度，用来控制列平滑程度。
 

【程序运行结果】

当修改 ax.plot_surface(X, Y, Z, rstride=3, cstride=3, cmap='rainbow') 中的参数 cmap='Accent' 及 cmap='bwr' 时，输出的三维图像分别如下所示。 

 
【参考文献】
https://blog.csdn.net/irykh/article/details/127964652
https://blog.csdn.net/Dontla/article/details/98616273