实验楼——Matplotlib 三维图形绘制方法

本节的学习内容比较简单，只是一些3维图形绘制的应用：

其中，主要用到的模块就是mpl_toolkits.mplot3d.axes3d()，其余模块的使用主要是plt内的各种方法。

1.三维点的绘制

import numpy as np
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
%matplotlib inline

# x, y, z 均为 0 到 1 之间的 100 个随机数
x = np.random.normal(0, 1, 100)
y = np.random.normal(0, 1, 100)
z = np.random.normal(0, 1, 100)
#生成画布
fig = plt.figure()
#声明3D对象
ax = Axes3D(fig)
ax.scatter(x, y, z)

2.生成3D曲面

# 创建 3D 图形对象
fig = plt.figure()
ax = Axes3D(fig)

# 生成数据
X = np.arange(-2, 2, 0.1)
Y = np.arange(-2, 2, 0.1)
X, Y = np.meshgrid(X, Y)
Z = np.sqrt(X ** 2 + Y ** 2)

# 绘制曲面图，并使用 cmap 着色
ax.plot_surface(X, Y, Z, cmap=plt.cm.winter)