Create Array Using Range Python