Hi there,
As long as it is not contraindicated (restricted due to wound or other issues) swimming is a great idea. There are specific rehabilitative measures that can be taken while in the pool which can be provided by a physiotherapist specializing in hydrotherapy.
Regards