draw a grid in python matplotlib