对数组进行按元素操作

Question

我正在尝试通过访问 Fortran 中的越界元素来检查我的数组是否返回无意义的内容。我想检查这些值是否小于 1，如果小于，请将其更改为 1。

这是我的代码片段引起的问题：

lastNeighLabel=(/clusterLabel(jj-1,kk,ll), clusterLabel(jj,kk-1,ll), clusterLabel(jj,kk,ll-1)/ ） LastNeighLabel 分别包含 x、y、z 方向上最后一个邻居的簇标签（介于 1 和 n 之间，其中 n 是找到的唯一单独簇的总数）。

当jj或kk或ll为1时，它们尝试访问数组中的第0个元素，并作为FORTRAN 从数组中的 1 开始计数，它试图摧毁宇宙。我目前正处于混乱的状态，大约有 8 个 if/elseif 语句，试图为每种可能性编写代码。但我希望有一种方法可以对每个元素进行操作。所以基本上我想说 where((/jj-1,kk-1,ll-1/).lt.1) do clusterLabel(jj-1,kk,ll)=0 等 取决于导致问题的元素。

但我想不出一种方法来做到这一点，因为 where 只会操作传递给它的变量，而不是同一索引处的不同数组。还是我错了？

如果这没有意义，我会很乐意编辑。

Answer 1

Fortran 并不强制从 1 开始访问数组。允许任何起始值。如果使用零索引数组更方便，请将数组声明为：

real, dimension (0:N-1, 0:M-1) :: array

或者

real, dimension (0:N, 0:M) :: array

并额外添加 0 索引以捕获特殊情况。

这可能是解决您问题的另一种解决方案，因为零索引值是合法的。

Answer 2

另一种可能的解决方法是创建一个扩展的簇标签数组（索引范围从 0 开始），它等于在外部附加一层零的簇标签数组。然后，您可以让循环在 jj、kk 和 ll 的所有值上安全运行。这是否是可行的解决方案取决于数组的大小。

integer :: extended_cluster_label(0:size(cluster_label,1),   &
                                  0:size(cluster_label,2),   &
                                  0:size(cluster_label,3)    &
                                 )

extended_cluster_label(0,:,:) = 0
extended_cluster_label(:,0,:) = 0
extended_cluster_label(:,:,0) = 0

extended_cluster_label(1:, 1:, 1:) = cluster_label

Answer 3

也许你可以使用一个函数？

  real function f(A,i,j,k)  
   real :: A(:,:,:)
   integer :: i,j,k

   if (i==0.or.j==0.or.k==0) then
    f=0
   else
    f=A(i,j,k)
   endif

  end function f

然后使用 f(clusterLabel,jj-1,kk,ll) 等。