在Python/Numpy/Scipy中寻找两个数组的插值交点

我明白你的需求——要基于两个在同一x支撑点上的数组，计算它们线性插值形成的分段直线之间的交点，而且要适配任意长度的数组。其实核心思路就是逐个区间检查相邻线段的相交情况，因为每个区间内的插值都是严格线性的，我们可以对每个区间单独计算交点。

核心逻辑拆解 #

对于每个连续的x区间 [x0, x1]：

• 第一条线段的端点是 (x0, y1[i]) 和 (x1, y1[i+1])
• 第二条线段的端点是 (x0, y2[i]) 和 (x1, y2[i+1])
• 我们需要判断这两条线段是否在区间内相交，如果相交，计算交点的x坐标

线性线段相交的计算可以通过解线性方程来实现：假设线段1的方程是 y = m1*(x - x0) + y1[i]，线段2是 y = m2*(x - x0) + y2[i]，令两者相等，解出x即可。需要注意斜率相同（平行）的情况，此时要么重合要么无交点。

完整代码实现 #

import numpy as np

def find_intersection_x(xSupport, y1, y2):
    # 初始化存储交点x坐标的列表
    intersections = []
    
    # 遍历每个x区间
    for i in range(len(xSupport) - 1):
        x0, x1 = xSupport[i], xSupport[i+1]
        y1_0, y1_1 = y1[i], y1[i+1]
        y2_0, y2_1 = y2[i], y2[i+1]
        
        # 计算两条线段的斜率，处理x区间长度为0的边界情况
        m1 = (y1_1 - y1_0) / (x1 - x0) if (x1 - x0) != 0 else np.inf
        m2 = (y2_1 - y2_0) / (x1 - x0) if (x1 - x0) != 0 else np.inf
        
        # 处理斜率相同的情况（平行或重合）
        if np.isclose(m1, m2):
            # 如果端点重合，记录对应x值（可根据需求选择是否保留）
            if np.isclose(y1_0, y2_0):
                intersections.append(x0)
            if np.isclose(y1_1, y2_1) and not np.isclose(x1, x0):
                intersections.append(x1)
            continue
        
        # 解线性方程求交点x
        x_intersect = x0 + (y2_0 - y1_0) / (m1 - m2)
        
        # 检查交点是否在当前x区间范围内
        if min(x0, x1) <= x_intersect <= max(x0, x1):
            intersections.append(x_intersect)
    
    # 转换为numpy数组返回，也可直接返回列表
    return np.array(intersections)

测试你的示例 #

xSupport = np.array([0,1])
y_1 = np.array([0,2])
y_2 = np.array([1,0])

print(find_intersection_x(xSupport, y_1, y_2))  # 输出: [0.33333333]，也就是1/3

额外说明 #

• 这个函数适配任意长度的输入数组，会返回所有区间内的交点x坐标
• 处理了x区间长度为0（x0=x1）的边界情况
• 对于平行线段，会记录端点重合的情况（如果不需要此逻辑可直接删除对应代码块）
• 返回结果为numpy数组，方便后续的数值处理

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