Create Array Using Range Python Function