Most ball pythons don't enjoy baths, they tolerate them. Also, being super-active is usually a sign that the snake isn't happy - it's trying to run away and hide.

If the vet recommended a bath then your ball python may have had a bit of stuck shed and was dehydrated. Also, if he had a ring of shed completely encircling his tail tip then yes, a quick dip to soften that shed so it doesn't get dried up and eventually cut off the blood flow to the tail isn't a bad idea. Long term though you need to fix your snake's humidity problems so it doesn't happen again.

I do have one snake that does enjoy swimming - she's a retic - getting her into a bath is easy, getting her out means getting soaked because she will fight to stay in.