Python实现3D曲面与形状的矢量图形可视化技术需求

Got it, let's solve your problem of creating true 3D vector graphics in Python that match the quality of LaTeX's pgfplots/tikz, with proper z-buffering for correct occlusion, and combining both surfaces and geometric shapes in a single plot.

The best tool for this job is matplotlib—it supports vector output formats (PDF/SVG) that behave just like pgfplots/tikz vectors, has built-in z-buffering for 3D rendering, and lets you mix multiple 3D elements seamlessly.

1. Install Dependencies #

First, make sure you have the required packages installed:

pip install matplotlib numpy

2. Full Code Example #

This code creates a 3D plot with a parabolic surface, a cube, and a sphere—all rendered with proper z-buffering, and saved as a vector PDF:

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d.art3d import Poly3DCollection

# Create figure and 3D axis (z-buffering is enabled by default)
fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(projection='3d')

# ----------------------
# 1. Draw 3D Surface (Paraboloid)
# ----------------------
x = np.linspace(-5, 5, 50)
y = np.linspace(-5, 5, 50)
X, Y = np.meshgrid(x, y)
Z = X**2 + Y**2  # Paraboloid equation

# Plot surface with a muted color to avoid overpowering other elements
ax.plot_surface(X, Y, Z, cmap='viridis', alpha=0.7, edgecolor='none')

# ----------------------
# 2. Draw Geometric Shape: Cube
# ----------------------
# Define cube vertices
cube_vertices = np.array([
    [-2, -2, 0], [2, -2, 0], [2, 2, 0], [-2, 2, 0],  # Bottom face
    [-2, -2, 8], [2, -2, 8], [2, 2, 8], [-2, 2, 8]   # Top face
])

# Define cube faces (each face is a list of vertex indices)
cube_faces = [
    [0, 1, 2, 3],  # Bottom
    [4, 5, 6, 7],  # Top
    [0, 1, 5, 4],  # Front
    [2, 3, 7, 6],  # Back
    [0, 3, 7, 4],  # Left
    [1, 2, 6, 5]   # Right
]

# Create polygon collection for the cube
cube_faces_polygons = [cube_vertices[face] for face in cube_faces]
cube = Poly3DCollection(cube_faces_polygons, facecolors='red', alpha=0.8, edgecolor='black')
ax.add_collection3d(cube)

# ----------------------
# 3. Draw Geometric Shape: Sphere
# ----------------------
u = np.linspace(0, 2 * np.pi, 30)
v = np.linspace(0, np.pi, 30)
sphere_x = 3 * np.outer(np.cos(u), np.sin(v))
sphere_y = 3 * np.outer(np.sin(u), np.sin(v))
sphere_z = 3 * np.outer(np.ones(np.size(u)), np.cos(v)) + 15  # Elevate sphere above the surface

# Plot sphere
ax.plot_surface(sphere_x, sphere_y, sphere_z, color='blue', alpha=0.6)

# ----------------------
# Configure Plot
# ----------------------
ax.set_xlabel('X Axis')
ax.set_ylabel('Y Axis')
ax.set_zlabel('Z Axis')
ax.set_title('3D Vector Plot with Surface + Geometric Shapes')
ax.set_zlim(0, 20)

# Save as VECTOR PDF (matches pgfplots/tikz vector quality)
plt.savefig('3d_vector_visualization.pdf', format='pdf', bbox_inches='tight', dpi=300)
plt.show()

• True Vector Graphics: Saving as PDF or SVG (replace format='pdf' with format='svg') produces fully scalable vector graphics—just like pgfplots/tikz. You can zoom in infinitely without pixelation, and edit elements in tools like Inkscape or Adobe Illustrator.
• Z-Buffering for Correct Occlusion: Matplotlib's 3D axis automatically handles z-buffering, so the cube is partially hidden by the paraboloid, and the sphere sits above both—no manual z-order adjustments needed (though you can tweak zorder if needed).
• Mixed 3D Elements: The plot combines a smooth surface, a polygonal cube, and a spherical surface all in one single figure, exactly as requested.

If you want to match LaTeX's typography perfectly, enable matplotlib's LaTeX rendering by adding this at the top of your code:

plt.rcParams.update({
    "text.usetex": True,
    "font.family": "serif"
})

This will make axis labels and titles use LaTeX fonts, aligning seamlessly with pgfplots-style output.

内容的提问来源于stack exchange，提问作者Quantum