如何提取Fortran字符串数组的子串？含编译报错示例

嘿，我来帮你搞定这个Fortran字符串数组提取子串的问题～你遇到的编译错误，核心原因是Fortran不支持直接用整数数组作为索引，对整个字符串数组批量截取子串——index(s,'_')返回的是和s同维度的整数数组，但s(1:i-1)这种写法试图用数组索引来批量操作每个字符串的子串，这在Fortran语法里是不允许的，因为子串的上下标必须是标量值。

下面给你几种可行的解决方案：

方法一：显式循环遍历数组元素 #

这是最直观的方式，逐个处理每个字符串元素，截取子串并转换类型（如果需要转整数的话）：

program testcharindex
implicit none
character(len=10), dimension(5) :: s
character(len=10), allocatable :: comp(:)
integer, allocatable :: i(:), n_int(:)
integer :: j

! 初始化字符串数组
s = (/ '1_E ', '2_S ', '3_E ', '14_E', '25_S' /)
! 分配对应大小的数组
allocate(i(size(s)), n_int(size(s)), comp(size(s)))

do j = 1, size(s)
    ! 获取每个字符串中'_'的位置
    i(j) = index(s(j),'_')
    ! 将下划线前的字符串转为整数（如果需要保留字符串可以直接赋值）
    read(s(j)(1:i(j)-1), *) n_int(j)
    ! 提取下划线后的子串
    comp(j) = s(j)(i(j)+1:)
end do

! 输出结果
print *,' i = ', i
print *,' n = ', n_int
print *,' comp = ', comp

! 释放分配的内存
deallocate(i, n_int, comp)
end program

注意：原代码里n定义为整数数组，但直接赋值字符串子串会触发类型不匹配错误，所以这里用read语句把字符串转成整数；如果只需要保留字符串形式，把n_int改成字符数组即可。

方法二：使用元素函数（Elemental Function） #

利用Fortran的elemental属性定义函数，函数会自动对数组的每个元素生效，实现批量处理，不用写显式循环：

program testcharindex
implicit none
character(len=10), dimension(5) :: s
character(len=10), allocatable :: comp(:)
integer, allocatable :: i(:), n_int(:)

! 定义元素函数：提取下划线前的部分并转为整数
elemental function get_prefix(str) result(res)
    character(len=*), intent(in) :: str
    integer :: res
    integer :: pos
    pos = index(str, '_')
    read(str(1:pos-1), *) res
end function get_prefix

! 定义元素函数：提取下划线后的子串
elemental function get_suffix(str) result(res)
    character(len=*), intent(in) :: str
    character(len=len(str)) :: res
    integer :: pos
    pos = index(str, '_')
    res = str(pos+1:)
end function get_suffix

! 初始化字符串数组
s = (/ '1_E ', '2_S ', '3_E ', '14_E', '25_S' /)
! 获取每个字符串中'_'的位置
i = index(s,'_')
! 批量提取前缀和后缀
n_int = get_prefix(s)
comp = get_suffix(s)

! 输出结果
print *,' i = ', i
print *,' n = ', n_int
print *,' comp = ', comp
end program

方法三：利用数组构造器+隐式循环 #

如果你的编译器支持Fortran 2003及以上标准，可以用数组构造器结合隐式循环来简化代码：

program testcharindex
implicit none
character(len=10), dimension(5) :: s
character(len=10), allocatable :: comp(:)
integer, allocatable :: i(:), n_int(:)

! 初始化字符串数组
s = (/ '1_E ', '2_S ', '3_E ', '14_E', '25_S' /)
! 获取每个字符串中'_'的位置
i = index(s,'_')

! 用隐式循环构造结果数组
n_int = [ (read(s(j)(1:i(j)-1), *), j=1, size(s)) ]
comp = [ (s(j)(i(j)+1:), j=1, size(s)) ]

! 输出结果
print *,' i = ', i
print *,' n = ', n_int
print *,' comp = ', comp
end program

这三种方法都能解决你遇到的问题，你可以根据自己的代码风格和编译器版本选择合适的方式～

内容的提问来源于stack exchange，提问作者gammarayon