如何利用一个数组做为索引，从另一个数组中取部分元素？

静待花开

如何利用一个数组做为索引，从另一个数组中取部分元素？

[Fortran] 纯文本查看 复制代码

program main
    implicit none
    integer :: A(5,3), B(3), C(3)
    A( 1, : ) = [ 5, 4, 8 ]
    A( 2, : ) = [ 4, 1, 9 ]
    A( 3, : ) = [ 3, 4, 8 ]
    A( 4, : ) = [ 1, 2, 8 ]
    A( 5, : ) = [ 3, 1, 0 ]
    B = [ 1, 3, 4 ]
    C = A(B)
    Write(6,*)A
    Write(6,*)B
    Write(6,*)C
end

编译后，提示下面错误 。

其实我就想从A中提取3个元素，这3个元素的索引是 1，3，4。将这三个元素存储在C中。不知为何总是提示错误。
我自己用一维数组A也做了测试，如下代码。编译执行都没有问题。请教高手，上面的代码为何编译提示错误呢？

[Fortran] 纯文本查看 复制代码

program main
    implicit none
    integer :: A(15), B(3), C(3)
    A = [5, 4, 8, 4, 1, 9, 3, 4, 8, 1, 2, 8, 3, 1, 0]
    B = [ 1, 3, 4 ]
    C = A(B)
    Write(6,*)A
    Write(6,*)B
    Write(6,*)C
end

fcode

只能一维数组这样。
多维（N维）数组因为每取一个数，需要N个数做为下标

如果第二维(其他维)的位置是固定的，可以这样：
C = A(B,1)
如果不是固定的，则不能这样，所以老老实实循环吧。

注意，不能写成 C=A(B,D)

[Fortran] 纯文本查看 复制代码

program main
  implicit none
  integer :: A(5,3), B(3),D(3), C(3) , j
  A( 1, : ) = [ 5, 4, 8 ]
  A( 2, : ) = [ 4, 1, 9 ]
  A( 3, : ) = [ 3, 4, 8 ]
  A( 4, : ) = [ 1, 2, 8 ]
  A( 5, : ) = [ 3, 1, 0 ]
  B = [ 1, 3, 4 ]
  D = [ 1, 2, 3 ]
  Do j = 1 , size(B)
    C(j) = A(B(j),D(j))
  End Do
  Write(6,*)C
end

li913

https://www.bilibili.com/video/BV1XD4y1S7jz?p=3  这里有介绍。

静待花开

fcode 发表于 2022-12-1 17:19
只能一维数组这样。
多维（N维）数组因为每取一个数，需要N个数做为下标

再次感谢您指点，您帮我回答过好几个问题了。非常感谢
看来我得花力气改代码了，
不知道您熟悉matlab不？有时候感觉这matlab与ofrtran的很多语法是相似的，不知道二者到底有没有联系。但是matlab里就可以实现二维数组作为索引，从一个大的数组里提取出这些索引位置对应的元素。

静待花开

li913 发表于 2022-12-1 18:00
https://www.bilibili.com/video/BV1XD4y1S7jz?p=3  这里有介绍。

非常感谢您的回复。这两个图也讲明白了，第2张图对我很有启发。看来二维还得老老实实注明第1维，第2维。