Lagrangian solutions for the semi-geostrophic shallow water system in physical space with general initial data