如何在Python中清晰展示R³中交错圆的前后遮挡关系

我是Python新手，想在3D空间中展示两个交错的圆，但Matplotlib绘制的图形无法正确呈现前后遮挡关系，希望实现类似Mathematica生成的交错遮挡效果，而非当前线条互相覆盖的结果。以下是我使用的代码，需要调整（视角后续自行处理）：

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

# Define the parametric equations for the circles
def circle_xy(t, h, k, r):
    x = h + r * np.cos(t)
    y = k + r * np.sin(t)
    z = 0 * np.ones_like(t)
    return x, y, z

def circle_yz(t, h, k, r):
    x = h + r * np.sin(t)
    y = 0 * np.ones_like(t)
    z = k + r * np.cos(t)
    return x, y, z

t = np.linspace(0, 2 * np.pi, 1000)

# Plotting
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

x1, y1, z1 = circle_xy(t, 0, 0, 1)
x2, y2, z2 = circle_yz(t, 1, 0, 1)

ax.plot(x1, y1, z1, label='unit circle centered at x=0, y=0 in the plane z=0',linewidth=3)
ax.plot(x2, y2, z2, label='unit circle centered at x=1, z=0 in the plane y=0',linewidth=3)

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

ax.view_init(30, 120)

plt.show()

问题根源 #

Matplotlib的3D绘图默认按绘制顺序渲染线条，不会自动根据空间位置判断前后遮挡关系，导致后绘制的线条会直接覆盖先绘制的线条，无论实际空间位置。

解决方案 #

手动根据每个点的空间深度，将线条分割为「被遮挡段」和「可见段」，按先画被遮挡部分、后画可见部分的顺序绘制。具体步骤为：

1. 计算每个点在当前视角下的深度值（通过3D坐标转屏幕投影坐标的z分量获取）
2. 对每个圆的线段按深度排序，从后到前依次绘制

修改后的代码如下：

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

def circle_xy(t, h, k, r):
    x = h + r * np.cos(t)
    y = k + r * np.sin(t)
    z = np.zeros_like(t)
    return x, y, z

def circle_yz(t, h, k, r):
    x = h + r * np.sin(t)
    y = np.zeros_like(t)
    z = k + r * np.cos(t)
    return x, y, z

t = np.linspace(0, 2 * np.pi, 1000)

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

# 获取两个圆的坐标
x1, y1, z1 = circle_xy(t, 0, 0, 1)
x2, y2, z2 = circle_yz(t, 1, 0, 1)

# 计算每个点的深度值（基于当前视角的投影）
def get_point_depth(x, y, z, ax):
    proj_matrix = ax.get_proj()
    # 将3D坐标转换为齐次坐标
    homogeneous_coords = np.vstack((x, y, z, np.ones_like(x))).T
    # 应用投影变换
    transformed_coords = homogeneous_coords @ proj_matrix.T
    # 返回投影后的z分量作为深度参考
    return transformed_coords[:, 2]

# 获取两个圆所有点的深度
depth_circle1 = get_point_depth(x1, y1, z1, ax)
depth_circle2 = get_point_depth(x2, y2, z2, ax)

# 按深度顺序绘制线段，实现遮挡效果
def plot_3d_with_occlusion(x, y, z, depth, ax, label, linewidth=3):
    # 生成所有线段的索引对
    segment_indices = np.array([(i, i+1) for i in range(len(x)-1)])
    # 计算每个线段的平均深度
    segment_depths = (depth[segment_indices[:, 0]] + depth[segment_indices[:, 1]]) / 2
    # 按深度从大到小排序（深度大的线段在后方，先绘制）
    sorted_seg_indices = np.argsort(segment_depths)[::-1]
    
    # 获取当前要使用的颜色
    line_color = ax._get_lines.get_next_color()
    # 按顺序绘制所有线段
    for idx in sorted_seg_indices:
        i, j = segment_indices[idx]
        ax.plot(x[i:j+1], y[i:j+1], z[i:j+1], color=line_color, linewidth=linewidth)
    # 绘制最后一个点与第一个点的闭合线段
    last_seg_depth = (depth[-1] + depth[0]) / 2
    ax.plot([x[-1], x[0]], [y[-1], y[0]], [z[-1], z[0]], color=line_color, linewidth=linewidth)
    # 添加图例项
    ax.plot([], [], label=label, color=line_color, linewidth=linewidth)

# 绘制两个圆，处理遮挡
plot_3d_with_occlusion(x1, y1, z1, depth_circle1, ax, 
                       'unit circle centered at x=0, y=0 in the plane z=0', linewidth=3)
plot_3d_with_occlusion(x2, y2, z2, depth_circle2, ax, 
                       'unit circle centered at x=1, z=0 in the plane y=0', linewidth=3)

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

ax.view_init(30, 120)

plt.show()

这段代码通过计算每个线段的深度并按从后到前的顺序绘制，让Matplotlib能正确呈现3D空间中的前后遮挡关系，接近你想要的效果。

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