python Axes3D绘制3D图形

# -*- coding: utf-8 -*-

#By:Eastmount CSDN

from matplotlib import pyplot as plt

import numpy as np

from mpl_toolkits.mplot3d import Axes3D  # 绘制3D坐标的函数

fig = plt.figure()               #创建一个绘图对象

ax = Axes3D(fig)                 #用这个绘图对象创建一个Axes对象

X = np.arange(-2, 2, 0.25)       #X轴-2到2之间

Y = np.arange(-2, 2, 0.25)       #Y轴-2到2之间

print(Y)

X, Y = np.meshgrid(X, Y)         #用两个坐标轴上的点在平面上画格

R = np.sqrt(X**2 + Y**2)         #X和Y的平方和开根号

Z = np.sin(R)                    #计算sin函数赋值为Z坐标

#具体函数方法可用 help(function) 查看，如：help(ax.plot_surface)

ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')

#给三个坐标轴注明

ax.set_xlabel('x label', color='r') 

ax.set_ylabel('y label', color='g') 

ax.set_zlabel('z label', color='b')

plt.show()