matplotlib——3D绘图

  • 开始
    • 绘制曲线
    • 绘制散点图
    • 改变图像角度

声明:

  1. 本文参考自matplotlib官方教程
  2. 介绍了3D曲线图和散点图的绘制,以后可能会添加更复杂图形的绘制方法。

开始

  使用matplotlib绘制3D图形,需要通过创建一个新的axes对象——Axes3D来实现。

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
plt.show()
   
   
   
   
  • 1
  • 2
  • 3
  • 4
  • 5

这里写图片描述

绘制曲线

  通过参数projection='3d来声明axes为axes3D,此时对axes3D对象调用plot()函数绘制曲线图时,多了如下表所示的几个参数,其他的参数还跟原来一样。

Argument Description
xs, ys x,y轴上的值
zs z轴上的值
zdir z轴的方向,可取{‘x’, ‘y’, ‘z’}
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.gca(projection='3d')

# Prepare arrays x, y, z
theta = np.linspace(-4 * np.pi, 4 * np.pi, 100)
z = np.linspace(-2, 2, 100)
r = z**2 + 1
x = r * np.sin(theta)
y = r * np.cos(theta)

ax.plot(x, y, z, label='parametric curve')  #这里传入x, y, z的值
ax.legend()

plt.show()
   
   
   
   

    这里写图片描述

    改变一下图像的角度:

    fig = plt.figure()
    ax = fig.add_subplot(1, 1, 1, projection='3d')
    for i in range(x.shape[0]):
        ax.text(x[i, 0], x[i, 1], x[i, 2], str(1), 
                color='r', 
                fontdict={'weight': 'bold', 'size': 9})
    ax.view_init(20, 0)  # 只有这一行改变了
    plt.show()
       
       
       
       

      你可能感兴趣的:(matplotlib——3D绘图)