Wi-Fi and similar communication protocols cause problems in case of emergencies, therefore are forbidden when the plane is on the ground. It's the same for take-off and landing. Once in mid air there are no emergencies that limit the use of such protocols hence Wi-Fi is turned on.