### 实现 Displacements 和 Mean Squared Displacement (MSD) 功能的 Python 代码示例 以下是一个完整的 Python 程序，用于从 LAMMPS 的 `dump` 文件中提取原子位置信息，并计算位移（Displacements）和均方位移（Mean Squared Displacement, MSD）。此代码利用 NumPy 进行高效的数值运算。 #### 数据准备 假设已经有一个名为 `dump.lammpstrj` 的 LAMMPS 轨迹文件，其中包含了模拟过程中的原子坐标信息。我们将从中读取数据并进行处理。 ```python import numpy as np def parse_lammps_dump(filename): """ 解析 LAMMPS dump 文件，返回每一帧中原子的位置数组。 """ frames = [] with open(filename, 'r') as file: lines = file.readlines() i = 0 while i < len(lines): if "ITEM: TIMESTEP" in lines[i]: timestep = int(lines[i + 1]) natoms = int(lines[i + 3].strip()) box_bounds = [ list(map(float, lines[i + 5].split())), list(map(float, lines[i + 6].split())), list(map(float, lines[i + 7].split())) ] atom_data_start = i + 9 frame_data = [] for j in range(atom_data_start, atom_data_start + natoms): _, x, y, z = map(float, lines[j].split()) frame_data.append([x, y, z]) frames.append(np.array(frame_data)) i += natoms + 9 else: i += 1 return np.array(frames) def calculate_displacements(positions): """ 计算每个原子相对于初始位置的位移向量。 """ initial_positions = positions[0] displacements = positions - initial_positions[np.newaxis, :] return displacements def calculate_msd(displacements): """ 根据位移向量计算均方位移 (MSD)。 """ squared_displacements = np.mean(np.sum(displacements**2, axis=2), axis=1) return squared_displacements # 主程序逻辑 if __name__ == "__main__": # 步骤 1：解析 LAMMPS dump 文件 filename = "dump.lammpstrj" all_frames = parse_lammps_dump(filename) # 步骤 2：计算位移 displacements = calculate_displacements(all_frames) # 步骤 3：计算均方位移 (MSD) msd_values = calculate_msd(displacements) # 输出结果 time_steps = np.arange(len(msd_values)) # 假设时间步长为索引本身 results = np.column_stack((time_steps, msd_values)) # 将结果保存到文件 np.savetxt("msd_results.txt", results, fmt="%.6f", header="TimeStep\tMSD") # 打印前几项结果供检查 print(results[:10]) # 显示前 10 行的结果 ``` --- ### 关键说明 1. **解析 LAMMPS Dump 文件** 上述代码定义了一个函数 `parse_lammps_dump`，它可以从标准的 LAMMPS `dump` 文件中逐帧提取原子坐标信息，并存储在一个三维 NumPy 数组中。该数组的第一维代表时间步长，第二维代表原子编号，第三维则对应空间坐标 \(x\)、\(y\)、\(z\)【^1】。 2. **计算位移** 函数 `calculate_displacements` 接收一个包含所有帧坐标的三维数组，并通过减去初始时刻的坐标矩阵得到每个原子在各时间步上的位移向量【^2】。 3. **计算均方位移 (MSD)** 函数 `calculate_msd` 利用位移向量计算平方和，并对其求平均值得到每一步的 MSD 值【^3】。 4. **输出与保存** 最终的 MSD 结果会被写入到一个文本文件 (`msd_results.txt`) 中，便于后续绘图或其他分析操作。 --- ###