Fortran 中返回数组的函数

Question

据我了解，您可以从 Fortran 中的函数返回一个数组，但由于某种原因，我的代码只返回我要求它返回的数组中的第一个值。这就是函数：

function polynomialMult(npts,x,y)
    integer npts
    double precision x(npts), results(npts + 1), y(npts,npts)

    polynomialMult =  x(1:npts) + 1

end function

这就是我

 C(1:numPoints) = polynomialMult(numPoints,x,f)

print *, C(1:numPoints)`

现在调用它的地方，它没有做任何有用的事情，因为我试图在编写逻辑之前理解语法。我看到了一些关于指定函数类型的内容，但是当我编写

integer function polynomialMult(npts,x,y)

或执行其他操作时，我收到编译错误。

Answer 1

要定义返回数组的函数，请在函数内部包含函数声明，如下所示：

function polynomialMult(npts,x,y)
    integer npts
    double precision x(npts), results(npts + 1), y(npts,npts)

! Change the next line to whatever you want
    double precision, dimension(npts) :: polynomialMult

    polynomialMult =  x(1:npts) + 1

end function

您的声明

integer function polynomialMult(npts,x,y)

声明该函数返回一个整数。 一个整数，而不是整数数组。我不认为标准允许函数声明，例如：

integer, dimension(10) function polynomialMult(npts,x,y)

但我可能是错的。我总是使用上面向您展示的表格。

如果您有最新的 Fortran 编译器，您可以做一些聪明的事情，例如返回分配的数组。我建议你弄清楚数组语法。例如，您的语句：

polynomialMult =  x(1:npts) + 1

可以更简洁地编写：

polynomialMult =  x + 1

因为 Fortran 会将标量加法映射到您声明为仅包含 npts 元素的数组 x 的所有元素。

将数组的大小传递到子例程中非常符合 FORTRAN77，而且现在几乎总是不必要的。通常，您要么想要对数组中的每个元素进行操作（如数组语法示例中所示），要么应该让子程序计算出它正在处理的数组的大小。

Answer 2

我同意前面的回复者的观点，即以下内容有效：

polynomialMult = x + 1

但是，如果不知道 polynomialMult 和 x 是数组，人们可能会认为它是标量运算。我更喜欢明显地这样做：

polynomialMult(:) = x(:) + 1

我什至坚持要求我小组中的编码员这样做。我不喜欢努力理解某人的代码——我希望他们在做什么是显而易见的。