Fortran Coder

查看: 2710|回复: 4

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

[复制链接]

68

帖子

23

主题

0

精华

熟手

F 币
244 元
贡献
132 点
发表于 2022-12-1 15:22:20 | 显示全部楼层 |阅读模式
如何利用一个数组做为索引,从另一个数组中取部分元素?

[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

编译后,提示下面错误 。
1.png
其实我就想从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




1948

帖子

12

主题

5

精华

论坛跑堂

臭石头雪球

F 币
1298 元
贡献
547 点

美女勋章热心勋章星光勋章新人勋章贡献勋章管理勋章帅哥勋章爱心勋章规矩勋章元老勋章水王勋章

发表于 2022-12-1 17:19:01 | 显示全部楼层
只能一维数组这样。
多维(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

790

帖子

2

主题

0

精华

大宗师

F 币
3765 元
贡献
2255 点
发表于 2022-12-1 18:00:11 | 显示全部楼层

评分

参与人数 1F 币 +3 收起 理由
fcode + 3 教科书级别~

查看全部评分

68

帖子

23

主题

0

精华

熟手

F 币
244 元
贡献
132 点
 楼主| 发表于 2022-12-1 18:34:39 | 显示全部楼层
fcode 发表于 2022-12-1 17:19
只能一维数组这样。
多维(N维)数组因为每取一个数,需要N个数做为下标

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

68

帖子

23

主题

0

精华

熟手

F 币
244 元
贡献
132 点
 楼主| 发表于 2022-12-1 18:38:04 | 显示全部楼层
li913 发表于 2022-12-1 18:00
https://www.bilibili.com/video/BV1XD4y1S7jz?p=3  这里有介绍。

非常感谢您的回复。这两个图也讲明白了,第2张图对我很有启发。看来二维还得老老实实注明第1维,第2维。
您需要登录后才可以回帖 登录 | 极速注册

本版积分规则

捐赠本站|Archiver|关于我们 About Us|小黑屋|Fcode ( 京ICP备18005632-2号 )

GMT+8, 2024-3-29 22:30

Powered by Tencent X3.4

© 2013-2024 Tencent

快速回复 返回顶部 返回列表