set shape method on Grid that returns self

It's convenient to set the shape in a chain of calls, like
g[g.something == 1800].set_shape((ny, nx)).contourf()