Изменение некоторую часть функции в модуле

Question

В # PSP_soil.py:

def evaporation_flux(psi): 
    h_s = exp(mw*psi/(R*T)) 
    return(E_p*(h_s-h_a)/(1-h_a)) 

Я хочу изменить эту функцию:

def evaporation_flux(psi): 
    h_s = exp(mw*psi/(R*T)) 
    return(h_s) 

но консоли в Spyder (Python 2.7) не запустите программу (E_p и h_a - постоянные переменные), и просто покажите, что UMD удалил: PSP_readDataFile, PSP_grid, PSP_ThomasAlgorithm, PSP_soil Любые советы в этом случае?

Answer 1

Вы можете сделать это:

from PSP_soil import * 

def evaporation_flux(psi): 
    h_s = exp(mw*psi/(R*T)) 
    return(h_s) 

Это переопределяет evaporation_flux из PSP_soil так, когда вы делаете evaporation_flux(value), она вызывается.

from PSP_soil import * импортирует все постоянные, необходимые для этой функции, но вы также можете сделать from PSP_soil import evaporation_flux, mv, R, T