计算两个指南针方向之间的差异Python

import math

def calculate_compass_difference(direction1, direction2):
    # 将方向转换为角度
    angle1 = direction_to_angle(direction1)
    angle2 = direction_to_angle(direction2)
    
    # 计算角度差异
    difference = angle2 - angle1
    
    # 处理角度差异超过360度的情况
    if difference > 180:
        difference -= 360
    elif difference < -180:
        difference += 360
    
    return difference

def direction_to_angle(direction):
    # 将方向转换为角度
    if direction == "北":
        return 0
    elif direction == "东":
        return 90
    elif direction == "南":
        return 180
    elif direction == "西":
        return 270
    else:
        raise ValueError("无效的方向")

# 示例用法
direction1 = "北"
direction2 = "东"
difference = calculate_compass_difference(direction1, direction2)
print("两个方向之间的差异为:", difference, "度")