Create Array With Range Python Function Return