Python中跳跃角（Jump angle）计算问题求助

针对转录因子二维轨迹的连续线段跳跃角计算问题，先分析现有实现的潜在问题，再给出更可靠的实现方式：

现有代码的问题 #

1. 未处理零向量：当轨迹中出现相邻点重合时，向量的模长为0，会触发除以零错误。
2. 数值精度隐患：通过点积+叉积计算带符号角度的方式，在极端情况（如向量几乎共线）下可能因浮点精度问题导致结果偏差。

改进后的实现 #

利用向量方向角的差值直接计算跳跃角，逻辑更直观且鲁棒性更强：

import numpy as np

def calculate_jump_angles(x_coords, y_coords):
    """计算轨迹中连续线段之间的跳跃角（带符号，范围[-π, π]）
    参数:
        x_coords: 轨迹点的x坐标数组
        y_coords: 轨迹点的y坐标数组
    返回:
        跳跃角列表，长度为轨迹点数量-2，异常情况（零向量）对应值为NaN
    """
    # 计算相邻点的位移向量
    dx = np.diff(x_coords)
    dy = np.diff(y_coords)
    
    # 计算每个位移向量的方向角（相对于x轴正方向，范围[-π, π]）
    dir_angles = np.arctan2(dy, dx)
    
    # 计算连续方向角的差值，即跳跃角
    jump_angles = np.diff(dir_angles)
    
    # 将角度归一化到[-π, π]范围（处理跨±π的跳变情况）
    jump_angles = np.mod(jump_angles + np.pi, 2 * np.pi) - np.pi
    
    # 标记零向量对应的无效跳跃角
    zero_vec_mask = (dx == 0) & (dy == 0)
    jump_angles[zero_vec_mask[:-1]] = np.nan
    jump_angles[zero_vec_mask[1:]] = np.nan
    
    return jump_angles.tolist()

关键说明 #

• 角度范围：默认返回带符号角度，逆时针转向为正，顺时针为负，范围[-π, π]。若需要[0, 2π]的无符号角度，可替换归一化步骤为jump_angles = np.mod(jump_angles, 2 * np.pi)。
• 异常处理：当轨迹中出现相邻点重合（零向量）时，对应位置的跳跃角设为NaN，便于后续数据过滤。
• 效率优化：使用numpy向量化运算，比循环实现更快，适合批量处理多条轨迹数据。

内容的提问来源于stack exchange，提问作者Yanisse Graïne