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