python:matplotlib:set_grid