Python简单物理计算代码异常结果的排查与修复求助

问题根源 #

Force2计算结果为0的核心原因是**forces数组的类型为整数型**，导致浮点数值被强制截断为0。

当你传入整数列表x=[1,2,3]时，np.zeros_like(x)会创建与x同类型的整数数组。此时将force(left_neighbour)返回的浮点值（约0.0659）赋值给forces[i]时，会被自动截断为整数0，最终显示结果与预期不符。

修复步骤 #

1. 修正forces数组的初始化类型 #

将函数中forces = np.zeros_like(x)替换为以下任意一种方式，确保数组为浮点类型：

# 方式1：直接指定浮点类型创建数组
forces = np.zeros(n, dtype=np.float64)

# 方式2：基于x的浮点版本创建数组
forces = np.zeros_like(np.asarray(x, dtype=np.float64))

2. 额外优化点 #

• 把dt作为参数传入函数（原代码未定义该变量，避免依赖全局变量）；
• 函数内部将输入的x和l统一转为浮点数组，避免后续计算中出现整数除法问题；
• 提前计算force_r和force_l，减少重复函数调用，同时让逻辑更清晰。

修复后的完整代码 #

import numpy as np

def force(r):
    if r == 0:
        return 0.0  # 返回浮点值，保持类型一致性
    else:
        return 1/r**13 - 1/r**7

def calculate_forces_and_positions(x, l, dt):
    # 统一转为浮点数组，避免整数类型问题
    x = np.asarray(x, dtype=np.float64)
    l = np.asarray(l, dtype=np.float64)
    n = len(x)
    
    # 创建浮点类型的forces数组
    forces = np.zeros(n, dtype=np.float64)
    
    for i in range(n):
        left_neighbour = (x[i] - x[i-1]) / l[i] if i > 0 else x[i] / l[i]
        right_neighbour = (x[i+1] - x[i]) / l[i] if i < n - 1 else 0.0
        
        force_r = force(right_neighbour)
        force_l = force(left_neighbour)
        forces[i] = -force_r + force_l
        
        print(f"n= {i} and L ={left_neighbour:.12f}  R = {right_neighbour:.12f} Force{i} = {forces[i]:.12f} force-R = {force_r:.12f} force-L= {force_l:.12f}")
    
    print(forces)
    x_new = x + dt * forces
    return forces, x_new

验证结果 #

传入x=[1,2,3]、l=[1,1,1]、dt=0.1调用函数后，n=2的Force2结果将显示为约0.065957928326，与公式推导的预期结果一致。

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