Fortran Coder

标题: 如何利用一个数组做为索引，从另一个数组中取部分元素？ [打印本页]

作者: 静待花开 时间: 2022-12-1 15:22
标题: 如何利用一个数组做为索引，从另一个数组中取部分元素？
如何利用一个数组做为索引，从另一个数组中取部分元素？

[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 时间: 2022-12-1 17:19
只能一维数组这样。
多维（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 时间: 2022-12-1 18:00
https://www.bilibili.com/video/BV1XD4y1S7jz?p=3 这里有介绍。

QQ截图20221201175804.png (189.32 KB, 下载次数: 199)

作者: 静待花开 时间: 2022-12-1 18:34

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

再次感谢您指点，您帮我回答过好几个问题了。非常感谢
看来我得花力气改代码了，

不知道您熟悉matlab不？有时候感觉这matlab与ofrtran的很多语法是相似的，不知道二者到底有没有联系。但是matlab里就可以实现二维数组作为索引，从一个大的数组里提取出这些索引位置对应的元素。

作者: 静待花开 时间: 2022-12-1 18:38

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

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

欢迎光临 Fortran Coder (http://bbs.fcode.cn/)