Generate An Array Of Numbers Python