使用python的matplotlib绘制三维函数图像

mac2024-10-11  58

示例程序如下:

import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D import numpy as np def fun(x, y): return x**2+y**2 fig1 = plt.figure() ax = Axes3D(fig1) X, Y = np.mgrid[-2:2:40j, -2:2:40j] Z = fun(X, Y) plt.title("This is main title") ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=plt.cm.coolwarm, alpha=0.5) ax.set_xlabel('x label', color='b') ax.set_ylabel('y label', color='r') ax.set_zlabel('z label', color='g') plt.show()
最新回复(0)