Fix x, y order with numpy.mgrid usage