可视化时序输入与输出|python

请帮我生成可视化图的python代码，输入是xxx变量，输出是xxx变量，横坐标是时间，输入用蓝线表示，输出用黄线表示，然后输入和输出在时间维度上是分别一个在前，一个在后。

import matplotlib.pyplot as plt

data = [1.310, 1.307, 1.307, 1.307, 1.307, 1.307, 1.307, 1.307, 1.307, 1.307]
output = [2.1, 2.5, 2.7]

time_input = range(len(data))
time_output = range(len(data), len(data) + len(output))
ground_truth = [1,2,3]


plt.plot(time_input, data, 'b-', label='Input')
plt.plot(time_output, output, 'y-', label='Output')
plt.plot(time_output, ground_truth, 'b-', label='GT')

plt.xlabel('Time')
plt.ylabel('Value')
plt.legend()

plt.show()