Python Initialize Array With Increasing Distance