Create Array Using Range Python Function Return