Question

我有一个关于在def

中提取变量的问题

例如，

def func(z,W):
    A = z[1]
    B = z[2]
    C = z[3]
    Var = k*B/C
    dAdW = A*3
    dBdW = B*2/Var
    dCdW = C**2*Var**2
    return dAdW, dBdW, dCdw
init = [0,0,1]
W = np.linspace(0,100,1000)
sol = odeint(func, init, W)
plt.plot(W,sol[:,0])
plt.plot(W,Var) # ----> this one

由于Var不是微分方程，

我不能将Var声明为Var = z [4]

我尝试使用全局但它只产生最后一个值。

如何在数组中获取Var的所有值，以便绘制Var？

k是一个常数，如1

感谢您查看此内容！

这是完整的代码：

import numpy as np
import matplotlib.pyplot as plt
from scipy.integrate import odeint

alpha = 0.0002
Uapb = 0.5
deltaH = -20000
Cpa = Cpb = 20
Cpi = 40
Ea = 25000
Kc1 = 100 # @ T=303
k1 = 0.004 #@ T=310K
CPcool = 20
Fao = 5
Cao = 1
mc = 1000
R = 1.987
To = 330


def func(z,W):
    pbar = z[0]
    X = z[1]
    Ta = z[2]
    T = z[3]
    k = k1*np.exp(Ea/R*(1/310-1/T))
    Kc = Kc1 * np.exp(deltaH / R * (1 / 303 - 1 / T))
    Ca = Cao * (1 - X) * To / T * pbar
    Cb = Cao*(1-X)*To/T*pbar
    Cc = 2*Cao*To/T*pbar
    Ci = To/T*pbar
    ra = k*(Ca*Cb-Cc**2/Kc)
    dxdw = -ra/Fao
    dpbardw = -alpha/(2*pbar)*T/To
    dTadw = (Uapb)*(T-Ta)/(mc*CPcool)
    dTdw = (Uapb*(Ta-T)+ra*deltaH)/(Fao*(Ca+Cb+Ci))
    return dxdw, dpbardw, dTadw, dTdw

init = [1, 0, 320, 320]
W = np.linspace(0,4500,1000)


sol = odeint(func, init, W)
#Xe = (2 * Kc - ((2 * Kc) ** 2 - 4 * Kc * (Kc - 4)) ** (1 / 2)) / (2 * (Kc - 4))
plt.plot(W, sol[:,0])
#plt.plot(W, Xe)

Answer 1

要捕获每次调用的数据，您可以执行以下操作：

capture_var = []

def func(z,W):
    Var = ...

    global capture_var
    capture_var.append(Var)

请注意：这是一个黑客，如果你做一些基本的脚本来解决一些小问题很可能。但是，如果你正在开发任何规模的代码库，请不要做这类事情，因为这些做法可能是维护的噩梦。

从python中的define函数中获取变量

1 个答案: