如何用Python绘制带周期性边界条件的粒子轨迹（解决跨边界连线问题）

我明白你遇到的问题——在周期性边界条件下，粒子从盒子一端出去、另一端进来时，直接用plt.plot()连线会画出一条横穿整个盒子的多余线条，完全不符合实际的运动轨迹。下面是针对性的解决方案：

核心思路 #

周期性边界下，当相邻时间步的粒子坐标差超过盒子边长的一半时，说明粒子是跨边界“瞬移”的，而非真的移动了这么远。我们需要针对每个粒子单独处理轨迹，在跨边界的位置断开连线，只绘制连续运动的片段。

修改后的代码 #

假设你的周期性盒子边长为Lx和Ly（请根据你的模拟参数修改），代码如下：

import numpy as np
import matplotlib.pyplot as plt

# 替换为你模拟中的周期性盒子边长
Lx = 1.0
Ly = 1.0

# 提取最近10个时间步的坐标
X_recent = X[time_steps-10:]
Y_recent = Y[time_steps-10:]
num_particles = X_recent.shape[1]

plt.figure(figsize=(12,11))
plt.grid(True)

# 逐个处理每个粒子的轨迹
for particle_idx in range(num_particles):
    # 获取当前粒子的x、y轨迹序列
    x_traj = X_recent[:, particle_idx]
    y_traj = Y_recent[:, particle_idx]
    
    # 检测跨边界的位置：相邻坐标差超过半边长即判定为跨边界
    cross_x = np.abs(np.diff(x_traj)) > Lx / 2
    cross_y = np.abs(np.diff(y_traj)) > Ly / 2
    # 合并x、y方向的跨边界点，得到需要断开的索引
    break_points = np.where(cross_x | cross_y)[0]
    
    # 拆分轨迹为连续片段并绘制
    start = 0
    for bp in break_points:
        # 绘制从start到bp的连续轨迹段
        plt.plot(x_traj[start:bp+1], y_traj[start:bp+1], 'k-')
        start = bp + 1
    # 绘制最后一段轨迹
    plt.plot(x_traj[start:], y_traj[start:], 'k-')

plt.show()

代码解释 #

1. 单独处理粒子轨迹：不再一次性绘制所有粒子的所有点，而是逐个粒子分析，确保轨迹控制精准。
2. 跨边界检测：通过计算相邻坐标的差值，判断粒子是否触发了周期性边界条件——差值超过半边长时，说明粒子是从边界“穿”过去的，而非连续移动。
3. 拆分绘制片段：把每个粒子的轨迹拆分成多个连续运动的片段，只在连续运动的点之间连线，彻底消除跨边界的多余线条。

如果你想要展示“展开式”的连续轨迹（而非盒子内的周期性轨迹），可以尝试将跨边界后的坐标进行平移（比如粒子从Lx出去时，下一个坐标加上Lx），但这种方式更适合观察轨迹的整体趋势，而非盒子内的实际位置。

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