错误使用 plot 向量长度必须相同。该怎么改

如果使用 plot 绘制多个向量时，向量长度不同会报错“向量长度必须相同”。此时应该将所有向量的长度调整为相同的长度，可以通过以下两种方式实现：

1. 对于长度不足的向量，在末尾添加元素，使其与长度最长的向量长度相同，可以使用 numpy 库中的 pad 函数实现：

import numpy as np
import matplotlib.pyplot as plt

# 原始向量
x = np.array([1, 2, 3])
y1 = np.array([4, 5])
y2 = np.array([6, 7, 8, 9])

# 将长度不足的向量末尾添加元素
y1 = np.pad(y1, (0, len(x)-len(y1)), 'constant', constant_values=(0))
y2 = np.pad(y2, (0, len(x)-len(y2)), 'constant', constant_values=(0))

# 绘制图像
plt.plot(x, y1)
plt.plot(x, y2)
plt.show()

2. 对于长度超过的向量，可以通过切片的方式截取相同长度的元素，使其与长度最短的向量长度相同：

import numpy as np
import matplotlib.pyplot as plt

# 原始向量
x1 = np.array([1, 2, 3, 4])
x2 = np.array([1, 2, 3])
y = np.array([4, 5, 6])

# 截取长度相同的元素
x1 = x1[:len(y)]
x2 = x2[:len(y)]

# 绘制图像
plt.plot(x1, y)
plt.plot(x2, y)
plt.show()