0

0

NumPy数组在循环中未更新:整数类型导致赋值截断问题

花韻仙語

花韻仙語

发布时间:2026-01-17 17:01:00

|

186人浏览过

|

来源于php中文网

原创

NumPy数组在循环中未更新:整数类型导致赋值截断问题

本文解释为何numpy数组在欧拉法求解微分方程的for循环中看似“不更新”,根本原因是数组默认为整数类型,导致浮点运算结果被强制截断为零,需显式声明为float类型。

在使用欧拉法(Euler’s method)数值求解常微分方程(如简谐振子系统)时,一个常见却隐蔽的错误是:数组元素在循环中始终不变化,即使右侧表达式已正确计算出非零浮点增量。你提供的代码正是典型场景——X[k+1] = X[k] + deriv(X[k], omega)*step 看似逻辑无误,但输出始终为 [0, 0],原因在于 X 的数据类型。

原始代码中:

X = np.asarray([[0 for _ in range(dimension)] for _ in t])

该语句创建的是 整数型(int64 或类似)二维数组。而 deriv() 函数返回的是浮点数组(例如 [1.0, -0.1]),当将其乘以 step=0.1 后得到如 [0.1, -0.01] 这样的小量;一旦尝试赋值给整数数组的某一行(如 X[k+1]),NumPy 会执行静默向下取整(truncation),即 0.1 → 0、-0.01 → 0,最终所有更新都被“抹平”为零。

✅ 正确做法是显式指定浮点类型

Text-To-Song
Text-To-Song

免费的实时语音转换器和调制器

下载
# 推荐写法:简洁、高效、语义清晰
X = np.zeros((len(t), dimension), dtype=float)

# 或等价地(兼容旧版本)
X = np.zeros((len(t), dimension), dtype=np.float64)
⚠️ 注意:np.asarray(...).astype(float) 虽可工作,但不如 np.zeros() 直接——后者避免了冗余的列表推导与类型转换,内存更优、速度更快。

修正后的完整可运行代码如下(仅改动初始化部分):

from matplotlib import pyplot as plt
import numpy as np

def deriv(X_k, omega):
    functions = [lambda _: X_k[1], lambda _: -omega**2 * X_k[0]]
    X_dot_k = np.array([f(1) for f in functions])
    return X_dot_k

step = 0.1
omega = 1
dimension = 2
t0, tf, x0, v0 = 0, 10, 1, 0
t = np.linspace(t0, tf, int((tf - t0) / step) + 1)

# ✅ 关键修复:使用 float 类型初始化
X = np.zeros((len(t), dimension), dtype=float)
X[0] = [x0, v0]

for k in range(len(t) - 1):
    X[k + 1] = X[k] + deriv(X[k], omega) * step

plt.plot(t, X[:, 0], label="position")
plt.xlabel("time (s)")
plt.ylabel("position (AU)")
plt.title("Position as a function of time (Euler method)")
plt.legend()
plt.grid(True)
plt.show()

? 补充建议:

  • 对于更高精度或更稳定求解,后续可升级为 scipy.integrate.solve_ivp;
  • 使用 print(X.dtype) 和 print(deriv(X[0], omega) * step) 可快速验证类型与中间值,是调试数值计算问题的黄金习惯;
  • 避免在循环中重复构造 lambda 列表(本例中可直接写为 np.array([X_k[1], -omega**2 * X_k[0]])),提升可读性与性能。

类型安全是科学计算的基石——一次 dtype 的疏忽,可能让整个数值模拟“静默失效”。

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python中print函数的用法
python中print函数的用法

python中print函数的语法是“print(value1, value2, ..., sep=' ', end=' ', file=sys.stdout, flush=False)”。本专题为大家提供print相关的文章、下载、课程内容,供大家免费下载体验。

184

2023.09.27

数据类型有哪几种
数据类型有哪几种

数据类型有整型、浮点型、字符型、字符串型、布尔型、数组、结构体和枚举等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

303

2023.10.31

php数据类型
php数据类型

本专题整合了php数据类型相关内容,阅读专题下面的文章了解更多详细内容。

222

2025.10.31

css中float用法
css中float用法

css中float属性允许元素脱离文档流并沿其父元素边缘排列,用于创建并排列、对齐文本图像、浮动菜单边栏和重叠元素。想了解更多float的相关内容,可以阅读本专题下面的文章。

559

2024.04.28

C++中int、float和double的区别
C++中int、float和double的区别

本专题整合了c++中int和double的区别,阅读专题下面的文章了解更多详细内容。

99

2025.10.23

lambda表达式
lambda表达式

Lambda表达式是一种匿名函数的简洁表示方式,它可以在需要函数作为参数的地方使用,并提供了一种更简洁、更灵活的编码方式,其语法为“lambda 参数列表: 表达式”,参数列表是函数的参数,可以包含一个或多个参数,用逗号分隔,表达式是函数的执行体,用于定义函数的具体操作。本专题为大家提供lambda表达式相关的文章、下载、课程内容,供大家免费下载体验。

204

2023.09.15

python lambda函数
python lambda函数

本专题整合了python lambda函数用法详解,阅读专题下面的文章了解更多详细内容。

190

2025.11.08

Python lambda详解
Python lambda详解

本专题整合了Python lambda函数相关教程,阅读下面的文章了解更多详细内容。

47

2026.01.05

高德地图升级方法汇总
高德地图升级方法汇总

本专题整合了高德地图升级相关教程,阅读专题下面的文章了解更多详细内容。

41

2026.01.16

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号