通过matplotlib绘制曲线的时候,当需要添加网格线的时候,在plt.show()之前加上plt.grid()即可。
import numpy as np
from collections import defaultdict, deque
import matplotlib.pyplot as plt
a = np.linspace(0, 5000, 100,endpoint=False)
b = np.sin(a)
plt.plot(a,b)
plt.xlabel('Episode Number')
plt.ylabel('Average Reward (Over Next %d Episodes)' % 100)
plt.grid()
plt.show()