Create Array With Range Python Function