python scipy integrate_python – 多种scipy.integrate.ode实例

待我称王封你为后i 2023-01-09 14:23 283阅读 0赞

一个选择是使用

multiprocessing(即使用进程而不是线程).这是一个使用multiprocessing.Pool类的map函数的示例.

函数求解需要一组初始条件,并返回由odeint生成的解.主要部分中的“串行”版本的代码重复调用解决方案,一次为ics中的每组初始条件. “多处理”版本使用multiprocessing.Pool实例的映射函数同时运行多个进程,每个调用解决.地图函数负责处理要解决的参数.

我的电脑有四个核心,当我增加num\_processes,加速最大值约3.6.

from \_\_future\_\_ import division, print\_function

import sys

import time

import multiprocessing as mp

import numpy as np

from scipy.integrate import odeint

def lorenz(q, t, sigma, rho, beta):

x, y, z = q

return \[sigma\*(y - x), x\*(rho - z) - y, x\*y - beta\*z\]

def solve(ic):

t = np.linspace(0, 200, 801)

sigma = 10.0

rho = 28.0

beta = 8/3

sol = odeint(lorenz, ic, t, args=(sigma, rho, beta), rtol=1e-10, atol=1e-12)

return sol

if \_\_name\_\_ == "\_\_main\_\_":

ics = np.random.randn(100, 3)

print("multiprocessing:", end='')

tstart = time.time()

num\_processes = 5

p = mp.Pool(num\_processes)

mp\_solutions = p.map(solve, ics)

tend = time.time()

tmp = tend - tstart

print(" %8.3f seconds" % tmp)

print("serial: ", end='')

sys.stdout.flush()

tstart = time.time()

serial\_solutions = \[solve(ic) for ic in ics\]

tend = time.time()

tserial = tend - tstart

print(" %8.3f seconds" % tserial)

print("num\_processes = %i, speedup = %.2f" % (num\_processes, tserial/tmp))

check = \[(sol1 == sol2).all()

for sol1, sol2 in zip(serial\_solutions, mp\_solutions)\]

if not all(check):

print("There was at least one discrepancy in the solutions.")

在我的电脑上,输出是：

multiprocessing: 6.904 seconds

serial: 24.756 seconds

num\_processes = 5, speedup = 3.59

发表评论取消回复

表情：

评论列表（有 0 条评论，283人围观）

还没有评论，来说两句吧...

相关阅读

相关 Python库使用：Numpy和SciPy的问题实例

在Python编程中，Numpy和SciPy是两个非常重要的科学计算库。以下是一个具体问题实例来说明它们如何使用： **问题实例：求解一元二次方程** 1. **Numpy

浅浅的花香味﹌/ 2025年02月11日 23:09/ 0 赞/ 72 阅读

相关使用Python进行科学计算：Numpy和Scipy的应用实例

在Python中，Numpy（Numeric Python）是一个用于处理大型多维数组的库。而Scipy（Scientific Computing Toolkit）则是在Num

谁践踏了优雅/ 2025年02月09日 18:36/ 0 赞/ 78 阅读

相关 python使用scipy.stats检验正态分布

from scipy import stats import numpy as np np.random.seed(1) x = stats.n

系统管理员/ 2023年10月10日 20:21/ 0 赞/ 110 阅读

相关 Python安装scipy包（python2.7）

Python安装scipy包（version：2.7） Python安装scipy包我们需要什么 numpy+mkl之外的包

傷城~/ 2023年06月10日 08:26/ 0 赞/ 90 阅读

相关 python scipy integrate_python – 多种scipy.integrate.ode实例

一个选择是使用 multiprocessing(即使用进程而不是线程).这是一个使用multiprocessing.Pool类的map函数的示例. 函数求解需要一组初始条件

待我称王封你为后i/ 2023年01月09日 14:23/ 0 赞/ 284 阅读

相关 Python安装scipy

直接使用Python的pip安装太慢了，30多M需要两个小时；使用清华源安装结果找不到这个库文件；豆瓣源可以，还可以指定版本。下面安装的是1.1.0版本因

拼搏现实的明天。/ 2022年11月01日 05:25/ 0 赞/ 313 阅读

相关 Python SciPy Sparse模块学习笔记

[原文地址][Link 1] 1. sparse模块的官方document地址：http://docs.scipy.org/doc/scipy/reference/spars

太过爱你忘了你带给我的痛/ 2022年06月15日 11:56/ 0 赞/ 325 阅读

相关 python的scipy实现插值

插值对于一些时间序列的问题可能比较有用。 Show the code directly: import numpy as np from matplotli

灰太狼/ 2022年06月01日 11:53/ 0 赞/ 295 阅读

相关 SciPy

官网地址：[https://scipy.org/][https_scipy.org] [博客][Link 1] SciPy（发音为“Sigh Pie”）是一个基

朱雀/ 2022年05月17日 05:34/ 0 赞/ 274 阅读

相关 python scipy.misc.imsave

为解决 [Error with scipy: No module named \`imsave\`][Error with scipy_ No module named _im

╰+攻爆jí腚メ/ 2022年02月26日 20:39/ 0 赞/ 315 阅读