python - 在 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()

这给了我一个由 2 条线组成的图表，每条线从x-axis = 0.

如何连接'a'并'b'绘制图表，以便在结束的地方'b'继续？x-axis'a'

谢谢！

标签： pythonnumpymatplotlib

Answer 1

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

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()