Matplotlib 绘制带箭头的相空间轨迹

Matplotlib 是 Python 中最流行的绘图库之一，用于创建高质量的图形。本文将介绍如何使用 Matplotlib 来绘制相空间轨迹，并给出完整的代码和示例。

什么是相空间？

相空间是描述物理系统状态演化的一种图像化方法，通常用一组变量的值表示。例如，物理学中常用的相空间变量是位置和动量。

如何绘制相空间轨迹？

绘制相空间轨迹通常需要在图上绘制多个箭头以表示状态的变化方向。下面是一个简单的例子，展示了如何使用 Matplotlib 绘制相空间轨迹。

首先，我们需要导入必需的库。

import matplotlib.pyplot as plt
import numpy as np

然后，定义一个函数来计算物理系统的状态演化。在这个例子中，我们将使用简单的二阶微分方程来描述状态的演变。

def func(y, t):
    return np.array([y[1], -y[0]])

接下来，定义一些初始条件，并用 odeint 函数来生成一组状态。

from scipy.integrate import odeint

# Initial condition
y0 = np.array([1.0, 0.0])
t = np.linspace(0, 10, 101)

# Solve ODE
y = odeint(func, y0, t)

现在，我们可以启动一个 Matplotlib 的绘图环境，并绘制相空间轨迹。

# Create a new figure
fig, ax = plt.subplots()

# Plot arrows
for i in range(0, len(t), 10):
    ax.arrow(y[i, 0], y[i, 1], y[i+1, 0]-y[i, 0], y[i+1, 1]-y[i, 1],
             head_width=0.05, head_length=0.1, fc='k', ec='k')

# Set axis limits
ax.set_xlim([-1.5, 1.5])
ax.set_ylim([-1.5, 1.5])

# Set axis labels
ax.set_xlabel('Position')
ax.set_ylabel('Momentum')

# Show the plot
plt.show()

图片示例：

            ..-^~~~^-.
          .~           ~.
         (;:           :;)
          (:           :)
            ':._   _.:'
                | |
              (=====)
                | |
                | |
                | |
              (=====)
                | |
                | |
              (=====)
                | |
          jgs   | |
               // \\
              //   \\
             //     \\
      ~~~~(_ (_))~~~~~

上述代码将生成一个相空间轨迹图，其中每一个箭头表示相空间状态的演化方向。在此示例中，我们使用了简单的二阶微分方程，但是 Matplotlib 可以用于绘制任意类型的相空间轨迹。