在不使用循环的情况下在 fortran 中用零填充数组

Question

我有两个数组，我想比较它们的大小并将尾随零添加到较短的数组。

例如-对于数组-

y1=(/ 1,2,3 /)
y2=(/ 1,2,3,4,5 /)

最后的结果应该是——

y1=(/ 1,2,3,0,0 /)
y2=(/ 1,2,3,4,5 /)

我对 Fortran 很陌生，据我所知，到目前为止，可以这样做：-

integer, allocatable :: y1(:),y2(:)
integer :: l1,l2,i
.
.
.
! some code to generate y1 and y2 here
.
.
.
l1=size(y1)
l2=size(y2)

if (l1>l2) then
    do i=l2+1,l1
        y2(i)=0
    enddo
else if (l2>l1) then
    do i=l1+1,l2
        y1(i)=0
    enddo
endif

我想知道是否有更好的方法来做到这一点，最好是不涉及循环的方法，因为我正在处理的实际问题可能有很大的向量

Answer 1

这是一种方法：

y1 = RESHAPE(y1,SHAPE(y2),pad=[0])

没有显式循环。正如@VladimirF 评论的那样，必须重新分配较短的数组，这种方法将它留给编译器和运行时来处理。

如果你正在关注这个方法的性能，或关心它的性能WRT使用显式循环的一个版本，并关注与数组的大小表现尺度如何，然后再运行一些测试。我不会惊讶地发现显式重新分配和一两个循环比这种“聪明”的方法更快。