Python Initialize Array With Increasing Altitude