ax.grid(axis='y') 
ax.grid(axis='x')