请帮我生成可视化图的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()