如何从 F2PY 中的回调函数中获取数组返回？

Question

我正在尝试使用 F2PY 编写一个从 Python 到 Fortran 的小接口，其中一个数组被传递给 Python 中的回调函数，结果数组被传递回 Fortran。 我有以下 Fortran 代码：

      Subroutine myscript(x,fun,o,n)
      external fun
      integer n
      real*8 x(n)
cf2py intent(in,copy) x
cf2py intent(out) o
cf2py integer intent(hide),depend(x) :: n=shape(x,0)
cf2py function fun(n,x) result o
cf2py integer intent(in,hide) :: n
cf2py intent(in), Dimension(n), depend(n) :: x
cf2py end function fun
      o = fun(n,x)
      write(*,*) o
      end

其中 fun 是 Python 中的回调函数，如下所示：

def f(x):
    print(x)
    return x

现在，当我使用 F2PY 包装 Fortran 代码并从 Python 运行它时，例如像这样：

myscript.myscript(numpy.array([1,2,3]),f)

我得到以下结果：

[1. 2. 3.]
1.00000000

显然，数组被传递给回调函数 f，但是当它被传回时，只有第一个条目被保留。 我需要做什么才能让整个阵列恢复原状？即在 Fortran 代码中获取变量 o 以包含数组 [1.,2.,3.] 而不是 1.?

Answer 1

好吧，我终于想通了。正如所指出的，o 必须被声明，然后o 也必须被放入函数fun 中。然后必须使用 Fortran 的 call 语句（而不是 o = fun(n,x)）调用该函数。显然，也可以摆脱大部分cf2py 语句。有趣的是，fun 不必显式声明为返回数组的函数。以下代码适用于我：

      Subroutine myscript(x,fun,o,n)
      external fun
      integer n
      real*8 x(n)
      real*8 o(n)
cf2py intent(in,copy), depend(n) :: x
cf2py intent(hide) :: n
cf2py intent(out), depend(n) :: o
      call fun(n,x,o)
      write(*,*) o
      end

返回

[1. 2. 3.]
1.00000000 2.00000000 3.00000000