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()