Question

我想使用单行语句在Fortran中执行以下伪代码的 do-end do 位：

integer, parameter :: N = 1000
integer, dimension(1:N) :: ArrayA, ArrayB
logical, dimension(1:N) :: ArrayL
...
...
do i = 1, N
    if( ArrayA(i) <= ArrayB(i) ) then
        ArrayL(i) = .true.
    else
        ArrayL(i) = .false.
    end if
end do

这可能吗？如果是这样，我该怎么做？

Answer 1

integer, parameter :: N = 1000
integer, dimension(1:N) :: ArrayA, ArrayB
logical, dimension(1:N) :: ArrayL
...
...
ArrayL = (ArrayA <= ArrayB)

Answer 2

@Mchinoune答案非常好。我发现WHERE / ELSEWHERE更具可读性，如果你以后进行操作，那么MASK很方便，并且可以直观地将其识别为MASK。

integer, parameter      :: N = 1000
integer, dimension(1:N) :: ArrayA, ArrayB
logical, dimension(1:N) :: The_Mask_Where_A_lt_B = .FALSE.
integer                 :: Some_MinVal
...
...
WHERE(ArrayA <= ArrayB) The_Mask_Where_A_lt_B = .TRUE.

!I know that this is not part of the question...
Some_MinVal = MINVAL(ArrayA, MASK=The_Mask_Where_A_lt_B)

Fortran中的矢量化数组比较

2 个答案: