如何用Python Matplotlib绘制3D空间中的无限直线（替代axline）

Matplotlib的3D坐标轴没有内置的axline函数，需要通过以下两种方式模拟无限直线的视觉效果：

方法一：生成超长线段（简单快捷） #

基于直线上的一点和方向向量，生成超出当前坐标轴范围的两个端点，绘制线段。Matplotlib会自动裁剪超出轴范围的部分，视觉上就呈现为无限延伸到边界的直线，缩放视图时只要线段长度足够，效果依然保持。

代码示例：

import matplotlib.pyplot as plt
import numpy as np

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

# 定义直线上的一点和方向向量
point = np.array([1, 1, 1])
direction = np.array([2, 3, 1])

# 生成超出轴范围的端点（scale可按需调整）
scale = 100
p_start = point - scale * direction
p_end = point + scale * direction

# 绘制黑色虚线
ax.plot([p_start[0], p_end[0]], [p_start[1], p_end[1]], [p_start[2], p_end[2]], 'k--')

# 设置坐标轴范围示例
ax.set_xlim(0, 5)
ax.set_ylim(0, 5)
ax.set_zlim(0, 5)

plt.show()

方法二：计算轴边界交点（精确贴合） #

通过计算直线与当前坐标轴边界平面的交点，筛选出在轴范围内的有效交点，取距离最远的两个作为端点，绘制的线段刚好贴合轴边界，没有冗余的超长部分。

代码示例：

import matplotlib.pyplot as plt
import numpy as np

def get_line_boundary_points(point, direction, xlim, ylim, zlim):
    points = []
    # 计算与x边界平面的交点
    if direction[0] != 0:
        for x_val in xlim:
            t = (x_val - point[0]) / direction[0]
            p = point + t * direction
            if ylim[0] <= p[1] <= ylim[1] and zlim[0] <= p[2] <= zlim[1]:
                points.append(p)
    # 计算与y边界平面的交点
    if direction[1] != 0:
        for y_val in ylim:
            t = (y_val - point[1]) / direction[1]
            p = point + t * direction
            if xlim[0] <= p[0] <= xlim[1] and zlim[0] <= p[2] <= zlim[1]:
                points.append(p)
    # 计算与z边界平面的交点
    if direction[2] != 0:
        for z_val in zlim:
            t = (z_val - point[2]) / direction[2]
            p = point + t * direction
            if xlim[0] <= p[0] <= xlim[1] and ylim[0] <= p[1] <= ylim[1]:
                points.append(p)
    # 取最远的两个交点，无交点则用超长线段兜底
    if len(points) >= 2:
        dists = [np.linalg.norm(p - point) for p in points]
        idx = np.argsort(dists)[-2:]
        return points[idx[0]], points[idx[1]]
    else:
        scale = 100
        return point - scale * direction, point + scale * direction

# 使用示例
fig = plt.figure()
ax = fig.add_subplot(projection='3d')

point = np.array([1, 1, 1])
direction = np.array([2, 3, 1])

# 设置坐标轴范围
xlim, ylim, zlim = (0, 5), (0, 5), (0, 5)
ax.set_xlim(xlim)
ax.set_ylim(ylim)
ax.set_zlim(zlim)

# 获取边界端点并绘制
p1, p2 = get_line_boundary_points(point, direction, xlim, ylim, zlim)
ax.plot([p1[0], p2[0]], [p1[1], p2[1]], [p1[2], p2[2]], 'k--')

plt.show()

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