Используя так называемые rc-параметры (глобальные настройки) можно задать вывод сетки для всех графиков так
plt.rcParams['axes.grid'] = True
На практике это можно сделать либо в заголовке файла, сразу после импорта:
import numpy as np
import matplotlib.pyplot as plt
plt.rcParams['axes.grid'] = True
Или в любом месте кода до вывода первого графика.