Python Define Array Of Numbers