所以我想学习f2py
,我有以下Fortran代码
subroutine fibonacci(a, n)
implicit none
integer :: i, n
double precision :: a(n)
do i = 1, n
if (i .eq. 1) then
a(i) = 0d0
elseif (i .eq. 2) then
a(i) = 1d0
else
a(i) = a(i - 1) + a(i - 2)
endif
enddo
end subroutine fibonacci
用f2py -c fibonacci.f -m fibonacci
编译并在Python中调用
import numpy
import fibonacci
a = numpy.zeros(13)
fibonacci.fibonacci(a)
print a
在Python中调用的子例程fibonacci
没有获得足够数量的参数,但代码神秘地起作用。顺便说一下,用fibonacci
调用子程序fibonacci.fibonacci(a, len(a))
也有效!
有人可以解释一下吗?谢谢!
f2py
知道a
和n
是函数参数,并且来自声明
double precision :: a(n)
它能够推断出n
是a
的长度。 NumPy数组有一个长度,因此Python包装器中不需要参数n
,而f2py
使它成为可选项。
请注意,f2py
生成的代码会检查您是否给出了太大的n
值:
In [19]: a = np.zeros(10)
In [20]: fibonacci.fibonacci(a, 99)
---------------------------------------------------------------------------
error Traceback (most recent call last)
<ipython-input-20-e9497469fd10> in <module>()
----> 1 fibonacci.fibonacci(a, 99)
error: (len(a)>=n) failed for 1st keyword n: fibonacci:n=99
您可以给出一个较小的值:
In [21]: a = np.zeros(10)
In [22]: fibonacci.fibonacci(a, 6)
In [23]: a
Out[23]: array([0., 1., 1., 2., 3., 5., 0., 0., 0., 0.])
您可能会发现生成并查看f2py
为此函数生成的接口文件很有用。命令
f2py -h fibonacci.pyf fibonacci.f
显示器
Reading fortran codes...
Reading file 'fibonacci.f' (format:fix,strict)
Post-processing...
Block: fibonacci
Post-processing (stage 2)...
Saving signatures to file "./fibonacci.pyf"
并生成包含的文件fibonacci.pyf
! -*- f90 -*-
! Note: the context of this file is case sensitive.
subroutine fibonacci(a,n) ! in fibonacci.f
double precision dimension(n) :: a
integer, optional,check(len(a)>=n),depend(a) :: n=len(a)
end subroutine fibonacci
! This file was auto-generated with f2py (version:2).
! See http://cens.ioc.ee/projects/f2py2e/
您可以从生成的声明中看到
integer, optional,check(len(a)>=n),depend(a) :: n=len(a)
f2py
推断n
应该是一个可选参数,其值不得超过a
的长度,其默认值为len(a)
。