如果你想要用三维的方式来画出这个函数的图像,你可以使用plot_surface
方法。首先,你需要从mpl_toolkits.mplot3d
包中导入Axes3D
。这样,你就可以创建一个三维的坐标轴,然后在这个坐标轴上画出你的图像。下面是一个示例代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
| import numpy as np import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D
x = np.linspace(-50, 50, 400) y = np.linspace(-50, 50, 400) X, Y = np.meshgrid(x, y) Z = 1/20 * X**2 + Y**2
fig = plt.figure() ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(X, Y, Z, cmap='viridis')
ax.set_xlabel('X') ax.set_ylabel('Y') ax.set_zlabel('Z') ax.set_title('Surface plot of 1/20*x^2 + y^2')
plt.show()
|
这段代码将会创建一个三维的曲面图,使用颜色来表示Z
值的大小。cmap='viridis'
是一个颜色映射选项,你可以根据自己的喜好更改。