静待花开 发表于 2022-12-1 15:22:20

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

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

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也做了测试,如下代码。编译执行都没有问题。请教高手,上面的代码为何编译提示错误呢?
program main
    implicit none
    integer :: A(15), B(3), C(3)
    A =
    B = [ 1, 3, 4 ]
    C = A(B)
    Write(6,*)A
    Write(6,*)B
    Write(6,*)C
end



fcode 发表于 2022-12-1 17:19:01

只能一维数组这样。
多维(N维)数组因为每取一个数,需要N个数做为下标

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

注意,不能写成 C=A(B,D):-(
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:11

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

静待花开 发表于 2022-12-1 18:34:39

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



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

静待花开 发表于 2022-12-1 18:38:04

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

非常感谢您的回复。这两个图也讲明白了,第2张图对我很有启发。看来二维还得老老实实注明第1维,第2维。
页: [1]
查看完整版本: 如何利用一个数组做为索引,从另一个数组中取部分元素?