Python轨道模拟故障：天体无法环绕运行的排查与修复方案

故障原因 #

1. 物理逻辑错误：将加速度作为标量累加进speed，再与初始速度简单合成，完全违背矢量运动规律——引力加速度是矢量，方向随天体相对位置实时变化，标量累加会导致速度方向和大小计算完全错误。
2. 角度未转弧度：初始速度方向veldir传入的是角度（270度），但math.cos()/math.sin()需要弧度参数，导致初始速度方向计算错误。
3. 循环条件无效：while (M1.x, M1.y) != (M2.x, M1.y)是错误逻辑，且浮点数位置几乎不可能完全相等，实际会导致循环直接终止或无意义执行。
4. pxM转换逻辑混乱：将加速度除以pxM，混淆了像素与真实单位的转换逻辑，应该在位置更新时处理单位转换，而非加速度阶段。
5. 冗余变量干扰：类中vel0、veldir、speed、fdir等变量设计冗余，导致速度合成逻辑混乱，不符合物理模拟的简洁性要求。

修复方案 #

1. 重构Body类：仅保留核心物理属性：位置(x,y)、速度分量(vx, vy)、质量、颜色，移除冗余变量。
2. 矢量化计算引力与加速度：直接计算引力的x/y分量，再转换为加速度分量，避免标量与方向分离的错误。
3. 添加固定时间步长：使用pygame.time.Clock()控制模拟的时间增量，保证模拟稳定性。
4. 修正角度转换：将初始速度角度转为弧度后计算速度分量。
5. 修正循环逻辑：改为无限循环，通过QUIT事件终止程序。

修正后的代码 #

import math
import pygame

# 像素与米的转换比例：1像素 = 3000米
PX_TO_M = 3000
# 模拟时间步长（秒）
TIME_STEP = 0.01

pygame.init()
width, height = 1920, 1080
screen = pygame.display.set_mode((width, height))
clock = pygame.time.Clock()

class Body:
    def __init__(self, mass, color, x, y, vx, vy):
        self.mass = mass
        self.color = color
        # 位置（像素）
        self.x = x
        self.y = y
        # 速度分量（米/秒）
        self.vx = vx
        self.vy = vy

    def render(self):
        pygame.draw.circle(screen, self.color, (int(self.x), int(self.y)), 5)

# 初始化天体：M1是大质量天体，M2是环绕天体
# M2初始速度：方向270度（向下），速度4像素/秒 → 转换为米/秒
initial_speed = 4 * PX_TO_M  # 像素/秒转米/秒
initial_dir_rad = math.radians(270)
M1 = Body(7e10, (255, 255, 255), 500, 500, 0, 0)
M2 = Body(1e6, (255, 0, 0), 700, 700, 
          initial_speed * math.cos(initial_dir_rad),
          initial_speed * math.sin(initial_dir_rad))

# 适配像素单位的引力常数（测试用）
G = 6.674e-11 * (PX_TO_M ** 3)

running = True
while running:
    # 控制帧率，获取时间步长
    dt = clock.tick(60) / 1000  # 转换为秒

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False

    # 计算天体间的相对位置
    dx = M2.x - M1.x
    dy = M2.y - M1.y
    distance_sq = dx ** 2 + dy ** 2
    distance = math.sqrt(distance_sq)

    # 避免距离为0的情况
    if distance < 1:
        break

    # 计算引力大小
    force_magnitude = G * M1.mass * M2.mass / distance_sq

    # 计算引力的分量
    fx = force_magnitude * dx / distance
    fy = force_magnitude * dy / distance

    # 计算加速度分量
    a1x = fx / M1.mass
    a1y = fy / M1.mass
    a2x = -fx / M2.mass
    a2y = -fy / M2.mass

    # 更新速度（米/秒）
    M1.vx += a1x * dt
    M1.vy += a1y * dt
    M2.vx += a2x * dt
    M2.vy += a2y * dt

    # 更新位置：速度（米/秒）转像素/秒，乘以时间步长
    M1.x += (M1.vx / PX_TO_M) * dt
    M1.y += (M1.vy / PX_TO_M) * dt
    M2.x += (M2.vx / PX_TO_M) * dt
    M2.y += (M2.vy / PX_TO_M) * dt

    # 渲染
    screen.fill((0, 0, 0))
    M1.render()
    M2.render()
    pygame.display.update()

pygame.quit()

说明 #

• 适配了像素与真实单位的转换，调整了G值以匹配模拟尺度；
• 采用矢量化物理计算，完全符合牛顿万有引力定律；
• 添加了固定时间步长，保证模拟的稳定性和可预测性；
• M2使用红色区分，便于观察环绕效果。

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