Creating An Array Of Numbers In Python