Your hair grafts at the recipient and donor areas will need the proper amount of time to heal and you should take every precaution to avoid any possible infections, so this does mean you should stay out of the water for at least 2 - 4 weeks postsurgery, or until Dr. Lao clears you for the activity.
Even after you’ve received an “Okay” to go swimming, you should always be mindful of the transplant donor area. There are also some diving positions, such as when the chin touches the chest or other “head down” positions, where the donor area could take a direct hit which will strain the area.