Entfernen von Versatzausdrücken aus einer Formel

Question

R verfügt über ein handliches Werkzeug zum Bearbeiten von Formeln, update.formula(). Dies funktioniert gut, wenn Sie etwas wie "Formel, die alle Begriffe in der vorherigen Formel außer x enthält", z.

f1 <- z ~ a + b + c 
(f2 <- update.formula(f1, . ~ . - c)) 
## z ~ a + b 

Allerdings scheint dies nicht mit Offset Bedingungen zu arbeiten:

f3 <- z ~ a + offset(b) 
update(f3, . ~ . - offset(b)) 
## z ~ a + offset(b) 

Ich habe so weit wie terms.formula gegraben nach unten, die ?update.formula Hinweise:

[nach Substitution , ...] Das Ergebnis wird dann vereinfacht über 'terms.formula (simplify = TRUE)'.

terms.formula(z ~ a + offset(b) - offset(b), simplify=TRUE) 
## z ~ a + offset(b) 

(dh dies scheint nicht offset(b) zu entfernen ...)

Ich weiß, dass ich eine Lösung zerhacken entweder durch deparse() und Textverarbeitung, oder durch die Formel Verarbeitung rekursiv, um den Begriff zu entfernen, den ich nicht möchte, aber diese Lösungen sind hässlich und/oder ärgerlich zu implementieren. Entweder Erleuchtung wie zu warum dies nicht funktioniert, oder eine einigermaßen kompakte Lösung, wäre toll ...

Answer 1

1) Rekursion Recursively durch die Formel rep absteigen Schnürung offset(...) mit offset und dann offset mit update entfernen. Es wird keine Zeichenfolgenmanipulation durchgeführt und obwohl es eine Anzahl von Codezeilen erfordert, ist es immer noch ziemlich kurz und entfernt einzelne und mehrere offset Begriffe.

Wenn mehrere Offsets vorhanden sind, können einige davon beibehalten werden, indem preserve festgelegt wird. Wenn beispielsweise preserve = 2 dann der zweite Offset beibehalten wird und alle anderen entfernt werden. Der Standardwert ist, keine zu behalten, d. H., Sie alle zu entfernen.

no.offset <- function(x, preserve = NULL) { 
    k <- 0 
    proc <- function(x) { 
    if (length(x) == 1) return(x) 
    if (x[[1]] == as.name("offset") && !((k<<-k+1) %in% preserve)) return(x[[1]]) 
    replace(x, -1, lapply(x[-1], proc)) 
    } 
    update(proc(x), . ~ . - offset) 
} 

# tests 

no.offset(z ~ a + offset(b)) 
## z ~ a 

no.offset(z ~ a + offset(b) + offset(c)) 
## z ~ a 

Beachten Sie, dass wenn Sie das preserve Argument nicht dann die Zeile Initialisierung k weggelassen werden kann und die if vereinfacht:

if (x[[1]] == as.name("offset")) return(x[[1]]) 

2) Begriffe diese Weder verwendet String-Manipulation direkt noch Rekursion.Zuerst erhalten Sie das terms Objekt, zap sein offset Attribut und beheben Sie es mit fixFormulaObject, die wir aus den Eingeweiden von terms.formula extrahieren. Dies könnte etwas weniger spröde gemacht werden, indem der Quellcode fixFormulaObject in die Quelle kopiert und die Zeile eval entfernt wird. preserve wirkt wie in (1).

no.offset2 <- function(x, preserve = NULL) { 
    tt <- terms(x) 
    attr(tt, "offset") <- if (length(preserve)) attr(tt, "offset")[preserve] 
    eval(body(terms.formula)[[2]]) # extract fixFormulaObject 
    f <- fixFormulaObject(tt) 
    environment(f) <- environment(x) 
    f 
} 

# tests 

no.offset2(z ~ a + offset(b)) 
## z ~ a 

no.offset2(z ~ a + offset(b) + offset(c)) 
## z ~ a 

Beachten Sie, dass, wenn Sie die preserve Argument nicht brauchen dann die Zeile, die die Offset-Attribut zappt kann vereinfacht werden:

attr(tt, "offset") <- NULL 

Answer 2

Dies scheint von Entwurf zu sein. Aber eine einfache Abhilfe ist

offset2 = offset 
f3 <- z ~ a + offset2(b) 
update(f3, . ~ . - offset2(b)) 
# z ~ a 

Wenn Sie die Flexibilität brauchen Formeln zu akzeptieren, die tut umfassen offset(), zum Beispiel, wenn die Formel durch ein Paket Benutzer zur Verfügung gestellt wird, die von der Notwendigkeit nicht bewusst sein kann offset2 an seinem Platz zu verwenden von offset, fügen Sie dann sollten wir auch eine Linie alle Instanzen von offset() in der eingehenden Formel zu ändern:

f3 <- z ~ a + offset(b) 

f4 <- as.formula(gsub("offset\\(", "offset2(", deparse(f3))) 
f4 <- update(f4, . ~ . - offset2(b)) 

# finally, just in case there are any references to offset2 remaining, we should revert them back to offset 
f4 <- as.formula(gsub("offset2\\(", "offset(", deparse(f4))) 
# z ~ a