我试图将python函数传递给f2py生成的函数。当python函数的输入和输出是一个数组时,我不知道如何做到这一点。(我是f2py的新手!)
这是我的fortran代码(num_Dg.f):
subroutine num_Dg(func,U,jacob,n)
double precision, intent(in) :: U(n)
double precision, intent(out) :: jacob(n,n)
double precision Up(n), Um(n)
double precision g_valp(n),g_valm(n)
double precision :: rel_eps = 2.e-5
double precision eps
integer i, j
external func
eps = minval(U)*rel_eps
do j=1,n
do i=1,n
if (i .eq. j) then
Up(i) = U(i)+(eps)
Um(i) = U(i)-(eps)
else
Up(i) = 0
Um(i) = 0
endif
enddo
call func(Up,g_valp,n)
call func(Um,g_valm,n)
do i=1,n
jacob(i,j) = 0.0
jacob(i,j) = (g_valp(i)-g_valm(i))/(2*eps)
enddo
enddo
end subroutine
我使用f2py编译:f2py -c num_Dg.f -m num_Dg
。它编译时没有错误。但是,当将函数导入python时,我得到以下结果:
>>> from num_Dg import num_dg
>>> print(num_dg.__doc__)
jacob = num_dg(func,u,[n,func_extra_args])
Wrapper for ``num_dg``.
Parameters
----------
func : call-back function
u : input rank-1 array('d') with bounds (n)
Other Parameters
----------------
func_extra_args : input tuple, optional
Default: ()
n : input int, optional
Default: len(u)
Returns
-------
jacob : rank-2 array('d') with bounds (n,n)
Notes
-----
Call-back functions::
def func(up,gvalp,[n]): return
Required arguments:
up : input rank-1 array('d') with bounds (n)
gvalp : input rank-1 array('d') with bounds (n)
Optional arguments:
n : input int, optional
Default: len(up)
回调函数func
有问题。输入应该是数组U
,输出应该是gval
。相反U
和gval
都是输入,没有输出
我假设这就是以下python脚本返回垃圾的原因:
import num_Dg
def func(x):
return [x[0]**2,x[1],x[2]]
val = [0.2,0.1,0.1]
num_Dg.num_dg(func,val)
如何更改fortran子例程,使回调函数正常工作?任何建议都将不胜感激!谢谢
我终于弄明白了。我认为它不起作用的原因是因为我错过了第
cf2py intent(out), depend(n) :: g_valp
行。可以使用f2py以正常方式编译脚本:f2py -c num_Dg.f -m num_Dg
相关问题 更多 >
编程相关推荐