如何通过变量赋值构建多维数组并实现原子-网格点距离计算

嘿，我来帮你搞定这个多原子与网格点的距离计算问题！

你现在要处理的是976个原子和2197个网格点的三维距离计算，之前处理33个原子时用的是单原子遍历网格点的写法，但原子数量翻了几十倍后，这种重复写多个原子的方式不仅代码冗余，效率也会拖后腿，对吧？

下面给你两种适合Fortran的解决方案，兼顾可读性和效率：

方案1：嵌套循环（清晰直白，容易调试） #

这种写法逻辑一目了然，外层循环遍历每个原子，内层循环遍历所有网格点，用一个二维数组存储所有距离结果——distances(i,j)代表第j个原子到第i个网格点的距离：

program calculate_atom_grid_distances
    implicit none
    ! 定义常量：原子数量和网格点数量
    integer, parameter :: NUM_ATOMS = 976, NUM_GRID = 2197
    ! 原子坐标数组（每个元素对应一个原子的X/Y/Z）
    real :: X(NUM_ATOMS), Y(NUM_ATOMS), Z(NUM_ATOMS)
    ! 网格点坐标数组（每个元素对应一个网格点的Xo/Yo/Zo）
    real :: Xo(NUM_GRID), Yo(NUM_GRID), Zo(NUM_GRID)
    ! 存储距离结果：行是网格点，列是原子
    real :: distances(NUM_GRID, NUM_ATOMS)
    integer :: atom_idx, grid_idx

    ! 这里先假设你已经完成了坐标数据的读入（比如从文件读取）

    ! 嵌套循环计算所有距离
    do atom_idx = 1, NUM_ATOMS
        do grid_idx = 1, NUM_GRID
            distances(grid_idx, atom_idx) = sqrt( &
                (X(atom_idx) - Xo(grid_idx))**2 + &
                (Y(atom_idx) - Yo(grid_idx))**2 + &
                (Z(atom_idx) - Zo(grid_idx))**2 &
            )
        end do
    end do

end program calculate_atom_grid_distances

如果后续计算（比如静电势）只需要距离的平方，可以直接去掉sqrt()，节省计算资源。

方案2：数组广播（简洁高效，利用Fortran数组特性） #

Fortran 2003及以后支持数组广播，不用写内层循环，代码更简洁，还能让编译器做自动优化：

program calculate_distances_broadcast
    implicit none
    integer, parameter :: NUM_ATOMS = 976, NUM_GRID = 2197
    real :: X(NUM_ATOMS), Y(NUM_ATOMS), Z(NUM_ATOMS)
    real :: Xo(NUM_GRID), Yo(NUM_GRID), Zo(NUM_GRID)
    real :: distances(NUM_GRID, NUM_ATOMS)

    ! 读入坐标数据...

    ! 用spread函数扩展数组维度，实现广播运算
    distances = sqrt( &
        (spread(X, 1, NUM_GRID) - spread(Xo, 2, NUM_ATOMS))**2 + &
        (spread(Y, 1, NUM_GRID) - spread(Yo, 2, NUM_ATOMS))**2 + &
        (spread(Z, 1, NUM_GRID) - spread(Zo, 2, NUM_ATOMS))**2 &
    )

end program calculate_distances_broadcast

spread(a, dim, n)会把一维数组a在指定维度dim上复制n次，让原子坐标和网格点坐标的维度匹配，直接完成逐元素运算。

• 如果需要结合电荷数组做后续计算（比如原子电荷在网格点产生的电势），可以直接在距离计算后叠加：potential(grid_idx, atom_idx) = charge(atom_idx) / distances(grid_idx, atom_idx)，两种方案都能轻松扩展。
• 大规模计算时，记得开启编译器优化选项（比如-O2或-O3），能大幅提升运行速度。

内容的提问来源于stack exchange，提问作者miguel.chavez13