如何在Python中对两条直线插值生成中间中心线？

嘿，这个需求其实挺常见的，而且根本不需要额外的Python库——用NumPy本身就能轻松实现！下面我给你分情况讲清楚怎么操作：

如果你的两个输入数组line1和line2都是形状为(n, 2)的NumPy数组（n是点的数量，每一行是一个(x, y)坐标），那中心线本质上就是对应点的坐标平均值。直接一行代码就能搞定：

import numpy as np

# 示例输入：两条各有3个点的折线
line1 = np.array([[0, 0], [2, 2], [4, 1]])
line2 = np.array([[1, 1], [3, 3], [5, 2]])

# 计算中心线：对应点坐标取平均
center_line = (line1 + line2) / 2

print(center_line)
# 输出：
# [[0.5 0.5]
#  [2.5 2.5]
#  [4.5 1.5]]

如果两条线的点数量不一样，我们可以先把其中一条线插值到和另一条相同的点数，再计算中心线。这里用NumPy的np.interp就能实现：

# 示例：line1有3个点，line2有5个点
line1 = np.array([[0, 0], [2, 2], [4, 1]])
line2 = np.array([[0, 1], [1, 2], [2, 3], [3, 2], [4, 1]])

# 提取line1的x坐标作为基准，对line2的y坐标进行插值
x_base = line1[:, 0]
x2, y2 = line2[:, 0], line2[:, 1]

# 插值得到和line1同点数的line2版本
y2_interpolated = np.interp(x_base, x2, y2)
line2_matched = np.column_stack((x_base, y2_interpolated))

# 现在可以计算中心线了
center_line = (line1 + line2_matched) / 2

如果想在两条线之间生成多条均匀分布的插值线，我们可以写个简单的函数来实现线性插值：

def generate_middle_lines(line1, line2, num_middle_lines=1):
    # 先确保两条线点数一致
    assert line1.shape == line2.shape, "两条线必须有相同数量的点哦"
    
    # 生成插值权重：从0到1之间取num_middle_lines个均匀值
    weights = np.linspace(0, 1, num_middle_lines + 2)[1:-1]
    
    # 计算每条中间线
    middle_lines = []
    for weight in weights:
        # 线性组合：line1*(1-weight) + line2*weight
        middle_line = line1 * (1 - weight) + line2 * weight
        middle_lines.append(middle_line)
    
    return middle_lines

# 示例：生成2条中间线
line1 = np.array([[0,0], [3,3]])
line2 = np.array([[0,3], [3,0]])
middle_lines = generate_middle_lines(line1, line2, num_middle_lines=2)

for idx, line in enumerate(middle_lines):
    print(f"第{idx+1}条中间线：")
    print(line)

如果你的“直线”其实是更复杂的曲线，或者需要基于垂直距离而非对应点来生成中心线，那可以试试shapely库（需要先安装：pip install shapely）。比如把两条线转为LineString，然后计算它们的中线，但这种场景相对少见，大多数情况下用NumPy的方法就足够了。

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