How can I create a 3D surface plot from a collection of points using Matplotlib?

Plotting Surfaces with Points in 3D Space using Matplotlib

In this article, we explore how to create a surface plot that encompasses a collection of points in three-dimensional space using Python's Matplotlib library, particularly its mplot3d module.

The plot_surface function in mplot3d requires two-dimensional arrays for X, Y, and Z arguments, rather than a list of 3-tuples as you have. Therefore, the first step is to prepare your data into the necessary format.

For surfaces, unlike line plots, you will need a 2D array grid representing the domain. Using discrete points, like the 3-tuples you have, presents a challenge because there are multiple potential triangulations that can create a surface.

Consider this Python code that generates a smooth surface, where f(x, y) = x^2 y:

<code class="python">import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

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)

# Calculate the Z values for each point in X and Y
zs = np.array(fun(np.ravel(X), np.ravel(Y)))
Z = zs.reshape(X.shape)

# Plot the surface
ax.plot_surface(X, Y, Z)

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

plt.show()</code>

In this example, X and Y are 2D arrays representing the domain, and Z is the corresponding array of values for each point. The plot_surface function uses these arrays to create a smooth surface. This approach is suitable for surfaces defined by a mathematical function.

However, if your data consists solely of discrete 3D points, you may need to consider other options.

The above is the detailed content of How can I create a 3D surface plot from a collection of points using Matplotlib?. For more information, please follow other related articles on the PHP Chinese website!