Matplotlib 3D绘图：Python数据可视化的新维度|极客教程

Matplotlib 3D绘图：Python数据可视化的新维度

参考：Three-dimensional Plotting in Python using Matplotlib

Matplotlib是Python中最流行的数据可视化库之一，它不仅能够创建二维图表，还能绘制令人印象深刻的三维图形。本文将深入探讨如何使用Matplotlib进行三维绘图，从基础概念到高级技巧，全面介绍这一强大功能。

1. 三维绘图的基础

在Matplotlib中，三维绘图主要依赖于mpl_toolkits.mplot3d模块。要开始三维绘图，首先需要创建一个带有3D投影的轴对象。

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 在这里添加绘图代码

plt.title('3D Plot Example - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

这段代码创建了一个基本的3D图形环境。fig.add_subplot(111, projection='3d')创建了一个3D轴对象，我们将在这个轴上进行所有的绘图操作。

2. 绘制三维散点图

散点图是最基本的三维图形之一，它可以直观地展示三维空间中的数据分布。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成随机数据
n = 100
x = np.random.rand(n)
y = np.random.rand(n)
z = np.random.rand(n)
colors = np.random.rand(n)
sizes = 1000 * np.random.rand(n)

scatter = ax.scatter(x, y, z, c=colors, s=sizes, alpha=0.6)

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.colorbar(scatter)
plt.title('3D Scatter Plot - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

在这个例子中，我们使用ax.scatter()函数绘制了一个三维散点图。每个点的位置由x、y、z坐标决定，颜色和大小可以用来表示额外的维度。alpha参数控制点的透明度，有助于在点重叠时更好地观察数据分布。

3. 绘制三维线图

三维线图可以用来展示三维空间中的路径或函数关系。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
t = np.linspace(0, 10, 100)
x = np.sin(t)
y = np.cos(t)
z = t

ax.plot(x, y, z, label='3D curve')

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')
ax.legend()

plt.title('3D Line Plot - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

这个例子展示了如何使用ax.plot()函数绘制三维线图。我们创建了一个螺旋线，它在x-y平面上旋转，同时在z轴方向上升高。

4. 绘制三维表面图

表面图是展示三维数据的另一种有效方式，特别适合表示二元函数。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x = np.linspace(-5, 5, 50)
y = np.linspace(-5, 5, 50)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))

# 绘制表面
surf = ax.plot_surface(X, Y, Z, cmap='viridis')

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

fig.colorbar(surf)
plt.title('3D Surface Plot - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

在这个例子中，我们使用ax.plot_surface()函数绘制了一个三维表面图。np.meshgrid()函数用于创建网格点，而cmap参数指定了颜色映射。

5. 绘制三维等高线图

等高线图可以在三维空间中展示相等值的轮廓线。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x = np.linspace(-5, 5, 50)
y = np.linspace(-5, 5, 50)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))

# 绘制等高线
contour = ax.contour(X, Y, Z, zdir='z', offset=-2, cmap='viridis')

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

fig.colorbar(contour)
plt.title('3D Contour Plot - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

这个例子使用ax.contour()函数在三维空间中绘制等高线。zdir参数指定等高线的投影方向，offset参数决定等高线在z轴上的位置。

6. 绘制三维柱状图

三维柱状图可以用来比较多个类别的数据。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x = np.arange(5)
y = np.arange(5)
z = np.random.randint(1, 20, size=(5, 5))

# 绘制柱状图
dx = dy = 0.8
dz = z
ax.bar3d(x.ravel(), y.ravel(), np.zeros_like(z).ravel(), dx, dy, dz.ravel(), shade=True)

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title('3D Bar Plot - how2matplotlib.com')
plt.show()

这个例子使用ax.bar3d()函数绘制三维柱状图。dx和dy参数控制柱子的宽度，dz参数决定柱子的高度。shade=True参数添加了阴影效果，增强了三维感。

7. 绘制三维线框图

线框图可以用来展示三维表面的结构。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x = np.linspace(-5, 5, 10)
y = np.linspace(-5, 5, 10)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))

# 绘制线框图
ax.plot_wireframe(X, Y, Z, color='r')

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title('3D Wireframe Plot - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

这个例子使用ax.plot_wireframe()函数绘制三维线框图。线框图可以清晰地展示表面的结构，特别是在复杂的三维形状中。

8. 绘制三维等值面

等值面是三维空间中具有相同值的点的集合，可以用来可视化三维标量场。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x, y, z = np.mgrid[-1:1:30j, -1:1:30j, -1:1:30j]
values = x*x + y*y + z*z

# 绘制等值面
ax.contour3D(x, y, z, values, 50, cmap='viridis')

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title('3D Isosurface Plot - how2matplotlib.com')
plt.show()

这个例子使用ax.contour3D()函数绘制三维等值面。等值面可以帮助我们理解三维空间中的数据分布和结构。

9. 绘制三维矢量场

矢量场可以用来表示三维空间中的流动或力场。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x, y, z = np.meshgrid(np.arange(-0.8, 1, 0.2),
                      np.arange(-0.8, 1, 0.2),
                      np.arange(-0.8, 1, 0.8))

u = np.sin(np.pi * x) * np.cos(np.pi * y) * np.cos(np.pi * z)
v = -np.cos(np.pi * x) * np.sin(np.pi * y) * np.cos(np.pi * z)
w = (np.sqrt(2.0 / 3.0) * np.cos(np.pi * x) * np.cos(np.pi * y) *
     np.sin(np.pi * z))

# 绘制矢量场
ax.quiver(x, y, z, u, v, w, length=0.1, normalize=True)

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title('3D Vector Field - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

这个例子使用ax.quiver()函数绘制三维矢量场。每个箭头的方向和长度由u、v、w分量决定。normalize=True参数确保所有箭头的长度相同。

10. 绘制三维曲面网格

曲面网格可以用来展示复杂的三维形状和结构。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
u = np.linspace(0, 2 * np.pi, 100)
v = np.linspace(0, np.pi, 100)
x = 10 * np.outer(np.cos(u), np.sin(v))
y = 10 * np.outer(np.sin(u), np.sin(v))
z = 10 * np.outer(np.ones(np.size(u)), np.cos(v))

# 绘制曲面网格
ax.plot_surface(x, y, z, rstride=4, cstride=4, color='b', alpha=0.7)

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title('3D Surface Mesh - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

这个例子使用ax.plot_surface()函数绘制三维曲面网格。rstride和cstride参数控制网格的密度，alpha参数调整透明度。

11. 绘制三维参数曲面

参数曲面是通过参数方程定义的三维曲面，可以用来表示复杂的几何形状。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
u = np.linspace(0, 2 * np.pi, 100)
v = np.linspace(-1, 1, 100)
U, V = np.meshgrid(u, v)
X = (1 + 0.5 * V * np.cos(U / 2)) * np.cos(U)
Y = (1 + 0.5 * V * np.cos(U / 2)) * np.sin(U)
Z = 0.5 * V * np.sin(U / 2)

# 绘制参数曲面
surf = ax.plot_surface(X, Y, Z, cmap='viridis')

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

fig.colorbar(surf)
plt.title('3D Parametric Surface - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

这个例子展示了如何绘制一个莫比乌斯带，这是一个著名的参数曲面。plot_surface()函数用于绘制曲面，而颜色映射则用于增强视觉效果。

12. 绘制三维隐函数曲面

隐函数曲面是由F(x,y,z)=0定义的三维曲面。这种曲面可以表示复杂的几何形状，如球体或环面。

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from skimage import measure

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x, y, z = np.mgrid[-2:2:100j, -2:2:100j, -2:2:100j]
values = x**2 + y**2 + z**2 - 1

# 计算等值面
verts, faces, _, _ = measure.marching_cubes(values, 0)

# 绘制隐函数曲面
ax.plot_trisurf(verts[:, 0], verts[:, 1], faces, verts[:, 2],
                cmap='viridis', lw=1)

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title('3D Implicit Surface - how2matplotlib.com')
plt.show()

这个例子展示了如何绘制一个由隐函数x^2 + y^2 + z^2 = 1定义的球体。我们使用skimage.measure.marching_cubes函数来计算等值面，然后用plot_trisurf函数绘制曲面。

13. 绘制三维极坐标图

极坐标系统在某些情况下比笛卡尔坐标系更适合表示某些数据。Matplotlib也支持在3D空间中绘制极坐标图。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
theta = np.linspace(0, 2*np.pi, 100)
r = 2 + np.sin(7*theta)
z = np.linspace(0, 2, 100)

# 转换为笛卡尔坐标
x = r * np.cos(theta)
y = r * np.sin(theta)

# 绘制3D极坐标图
ax.plot(x, y, z, label='3D Polar curve')

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')
ax.legend()

plt.title('3D Polar Plot - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

这个例子展示了如何在3D空间中绘制一个极坐标曲线。我们首先在极坐标系中定义曲线，然后将其转换为笛卡尔坐标系进行绘制。

14. 绘制三维直方图

三维直方图可以用来可视化三维数据的分布情况。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x = np.random.normal(0, 1, 1000)
y = np.random.normal(0, 1, 1000)
z = np.random.normal(0, 1, 1000)

# 绘制3D直方图
hist, xedges, yedges = np.histogram2d(x, y, bins=20, range=[[-3, 3], [-3, 3]])
xpos, ypos = np.meshgrid(xedges[:-1] + 0.25, yedges[:-1] + 0.25, indexing="ij")
xpos = xpos.ravel()
ypos = ypos.ravel()
zpos = 0

dx = dy = 0.5 * np.ones_like(zpos)
dz = hist.ravel()

ax.bar3d(xpos, ypos, zpos, dx, dy, dz, zsort='average')

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Frequency')

plt.title('3D Histogram - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

这个例子展示了如何创建一个三维直方图。我们首先生成三维正态分布的数据，然后使用np.histogram2d计算二维直方图，最后用bar3d函数绘制三维柱状图。

15. 绘制三维等高线填充图

等高线填充图可以用来展示三维数据的层次结构。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x = np.linspace(-5, 5, 50)
y = np.linspace(-5, 5, 50)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))

# 绘制等高线填充图
cset = ax.contourf(X, Y, Z, zdir='z', offset=-2, cmap='viridis')

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

fig.colorbar(cset)
plt.title('3D Filled Contour Plot - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

这个例子使用ax.contourf()函数绘制三维等高线填充图。与之前的等高线图不同，这里的等高线之间的区域被填充了颜色，更好地展示了数据的变化趋势。

16. 绘制三维流线图

流线图可以用来可视化向量场，特别适合展示流体动力学或电磁场等物理现象。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x, y, z = np.meshgrid(np.arange(-0.8, 1, 0.2),
                      np.arange(-0.8, 1, 0.2),
                      np.arange(-0.8, 1, 0.8))

u = np.sin(np.pi * x) * np.cos(np.pi * y) * np.cos(np.pi * z)
v = -np.cos(np.pi * x) * np.sin(np.pi * y) * np.cos(np.pi * z)
w = (np.sqrt(2.0 / 3.0) * np.cos(np.pi * x) * np.cos(np.pi * y) *
     np.sin(np.pi * z))

# 绘制流线图
ax.streamplot(x, y, u, v, density=1, linewidth=1, arrowsize=1, arrowstyle='->')

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title('3D Streamplot - how2matplotlib.com')
plt.show()

这个例子使用ax.streamplot()函数绘制三维流线图。流线的密度、线宽和箭头样式都可以自定义。

17. 绘制三维热图

热图是一种用色彩来表示数值大小的图表，在三维空间中可以用来展示复杂的数据分布。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x = y = np.arange(-3.0, 3.0, 0.05)
X, Y = np.meshgrid(x, y)
Z = np.exp(-(X**2 + Y**2))

# 绘制热图
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='hot', linewidth=0, antialiased=False)

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

fig.colorbar(surf, shrink=0.5, aspect=5)
plt.title('3D Heatmap - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度

这个例子使用ax.plot_surface()函数绘制三维热图。我们使用了’hot’颜色映射来表示数值的大小，红色表示高值，蓝色表示低值。

18. 绘制三维等值线图

等值线图在三维空间中可以用来展示复杂函数的特性。

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

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

# 生成数据
x = np.arange(-5, 5, 0.25)
y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(x, y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)

# 绘制等值线图
ax.contour(X, Y, Z, zdir='z', offset=-2, cmap='coolwarm')

ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title('3D Contour Plot - how2matplotlib.com')
plt.show()

Output:

Matplotlib 3D绘图：Python数据可视化的新维度