python - matplotlib 中的曲面图

Question

我有一个 3 元组列表，表示 3D 空间中的一组点。我想绘制一个覆盖所有这些点的曲面。

mplot3d 包中的 plot_surface 函数要求参数 X、Y 和 Z 为二维数组。 plot_surface 是绘制表面的正确函数吗？如何将数据转换为所需的格式？

data = [(x1,y1,z1),(x2,y2,z2),.....,(xn,yn,zn)]

Answer 1

对于表面，它与 3 元组列表有点不同，您应该在 2d 数组中为域传递一个网格。

如果你只有一个 3d 点列表，而不是某个函数 f(x, y) -> z，那么你就会遇到问题，因为有多种方法可以对 3d 进行三角测量点云到表面。

这是一个光滑表面的例子:

import numpy as np
from mpl_toolkits.mplot3d import Axes3D  
# Axes3D import has side effects, it enables using projection='3d' in add_subplot
import matplotlib.pyplot as plt
import random

def fun(x, y):
    return x**2 + y

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
x = y = np.arange(-3.0, 3.0, 0.05)
X, Y = np.meshgrid(x, y)
zs = np.array(fun(np.ravel(X), np.ravel(Y)))
Z = zs.reshape(X.shape)

ax.plot_surface(X, Y, Z)

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

plt.show()