在 matplotlib 上绘制 numpy 数组

Question

我有以下代码：

import matplotlib.pyplot as plt
import numpy as np
a=np.array([[0],[1],[2]], np.int32)
b=np.array([[3],[4],[5]], np.int32)

plt.plot(a, color = 'red', label = 'Historical data')
plt.plot(b, color = 'blue', label='Predicted data')
plt.legend()
plt.show()

这给了我一个从x-axis = 0开始的 2 条线图。

如何连接'a'和'b'和 plot 图形，使'b'在x-axis继续'a'结束？

谢谢！

Answer 1

您可以添加一个x数组，然后在下一个 plot 中增加其值，以便将其附加到前一个 plot 中。

import matplotlib.pyplot as plt
import numpy as np
a=np.array([[0],[1],[2]], np.int32)
b=np.array([[3],[4],[5]], np.int32)
x = np.arange(a.shape[0])

plt.plot(x, a, color = 'red', label = 'Historical data')
plt.plot(x+a.shape[0], b, color = 'blue', label='Predicted data')
plt.legend()
plt.show()