pythonbook/实例学习Numpy与Matplotlib/matplotlib 三维模型.py

15 lines
305 B
Python

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np
from matplotlib import cm
fig=plt.figure()
ax=fig.add_subplot(111,projection='3d')
u=np.linspace(-1,1,100)
x,y=np.meshgrid(u,u)
z=x**2+y**2
ax.plot_surface(x,y,z,rstride=4,cstride=4,cmap=cm.YlGnBu_r)
plt.show()