python 计算任意两个角度/弧度的差值

def angle_diff(angle_1, angle_2):
    """
    输入弧度
    """
    angle_1 = angle_1 % (2*np.pi)
    angle_2 = angle_2 % (2*np.pi)
    
    angle_min = min(angle_1, angle_2)
    angle_max = max(angle_1, angle_2)

    error = angle_max - angle_min
    if error <= np.pi:
        return error
    else:
        return angle_min + 2*np.pi - angle_max