Fix x, y order in np.mgrid usage