wind data python code