python 绘图---2D、3D散点图、折线图、曲面图( 二 ) _绘制

文章插图
4. 绘制3D图曲面图
()用于绘制3D曲面图：

文章插图

# -*- coding: utf-8 -*-from matplotlib import pyplot as pltimport numpy as npfrom mpl_toolkits.mplot3d import Axes3Dfig = plt.figure()ax = Axes3D(fig)X = [0,1.1,1.8,3.1,4.0]Y = [2,2.4,4.3,3.5,2.5]X, Y = np.meshgrid(X, Y)Z = np.array([[2,2.4,4.3,3.5,2.5]])ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')plt.draw()plt.pause(10)plt.savefig('3D.jpg')plt.close()

绘图：

文章插图

文章插图
使用()绘制另一个3D曲面：

# -*- coding: utf-8 -*-from matplotlib import pyplot as pltimport numpy as npfrom mpl_toolkits.mplot3d import Axes3Dfig = plt.figure()ax = Axes3D(fig)X = np.arange(-4, 4, 0.25)Y = np.arange(-4, 4, 0.25)X, Y = np.meshgrid(X, Y)R = np.sqrt(X**2 + Y**2)Z = X*Yax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')plt.draw()plt.pause(10)plt.savefig('3D.jpg')plt.close()

绘图：

文章插图

文章插图
5. 绘制3D散点图
()用于绘制3D散点图：

# -*- coding: utf-8 -*-import numpy as npimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3Dx = [0,1.1,1.8,3.1,4.0]y = [2,2.4,4.3,3.5,2.5]z = [1,2,3,4,5]ax = plt.subplot(projection='3d')# 创建一个三维的绘图工程ax.scatter(x, y, z, c='r')# 绘制数据点,颜色是红色ax.set_zlabel('Z')# 坐标轴ax.set_ylabel('Y')ax.set_xlabel('X')plt.draw()plt.pause(10)plt.savefig('3D.jpg')plt.close()

绘图：

文章插图
【python 绘图---2D、3D散点图、折线图、曲面图】

文章插图