Come scrivere una funzione R che valuta l'espressione all'interno di un data-frame

Question

Puzzle per intenditori R: Supponiamo di avere un data-frame:

df <- data.frame(a = 1:5, b = 1:5) 

So che possiamo fare le cose come

with(df, a) 

per ottenere un vettore di risultati.

Ma come faccio a scrivere una funzione che prende un'espressione (come a o a > 3) e fa la stessa cosa dentro. Cioè Voglio scrivere una funzione fn che accetta un quadro dati e un'espressione come argomenti e restituisce il risultato della valutazione dell'espressione "all'interno" del frame dei dati come ambiente.

Non importa se questo sembra forzato (potrei semplicemente usare with come sopra), ma questa è solo una versione semplificata di una funzione più complessa che sto scrivendo. Ho provato diverse varianti (usando eval, with, envir, substitute, local, ecc.) Ma nessuna di queste funzioni. Per esempio, se io definisco fn in questo modo:

fn <- function(dat, expr) { 
    eval(expr, envir = dat) 
} 

ottengo questo errore:

> fn(df, a) 
Error in eval(expr, envir = dat) : object 'a' not found 

Chiaramente mi manca qualcosa di sottile sugli ambienti e valutazione. C'è un modo per definire una funzione del genere?

Answer 1

Il pacchetto reticolo fa questo genere di cose in un modo diverso. Vedi, ad esempio, lattice:::xyplot.formula.

fn <- function(dat, expr) { 
    eval(substitute(expr), dat) 
} 
fn(df, a)    # 1 2 3 4 5 
fn(df, 2 * a + b)  # 3 6 9 12 15 

Answer 2

Questo perché non stai passando un'espressione.

Prova:

fn <- function(dat, expr) { 
    mf <- match.call() # makes expr an expression that can be evaluated 
eval(mf$expr, envir = dat) 
} 

> df <- data.frame(a = 1:5, b = 1:5) 
> fn(df, a) 
[1] 1 2 3 4 5 
> fn(df, a+b) 
[1] 2 4 6 8 10 

Una rapida occhiata al codice sorgente delle funzioni utilizzare questo (ad esempio lm) può rivelare molto cose più interessanti su di esso.

Answer 3

? All'interno potrebbe anche essere di interesse.

df <- data.frame(a = 1:5, b = 1:5) 
within(df, cx <- a > 3) 
    a b cx 
1 1 1 FALSE 
2 2 2 FALSE 
3 3 3 FALSE 
4 4 4 TRUE 
5 5 5 TRUE 

Answer 4

Una voce in ritardo, ma l'approccio e la sintassi data.table sembrerebbe essere quello che sono dopo. Questo è esattamente il modo in cui [.data.table funziona con gli argomenti j, i e by.

Se ne avete bisogno in forma fn(x,expr), quindi è possibile utilizzare il seguente

library(data.table) 

DT <- data.table(a = 1:5, b = 2:6) 

`[`(x=DT, j=a) 

## [1] 1 2 3 4 5 

`[`(x=DT, j=a * b) 
## [1] 2 6 12 20 30 

penso che sia più facile da usare in forma più nativa

DT[,a] 
## [1] 1 2 3 4 5 

e così via. Sullo sfondo si utilizza substitute e eval