使用matplotlib在更新图中链接点的Python线

Question

我正在使用matplotlib在强化学习应用程序中绘制奖励与情节图。因此，基本上，每次情节结束时我都会更新情节。 在这里python-How do I plot in real time in a while loop using matplotlib进行搜索，我可以在剧集每次完成时使剧情更新。我的情节代码是：

reward_plot = plt.plot(i,total_reward, '-ro')
plt.title('Reward vs episodes')
plt.xlabel("Episodes", fontsize=12)
plt.ylabel("Reward", fontsize=12)
plt.pause(0.05)

情节实际上看起来像这样：

我只想让绘图在每次更新图形时画一条连接点的线。有可能的？我尝试了不同的方法，但是没有运气。我正在使用Python 3.6.6。