Làm thế nào để viết một chức năng R mà đánh giá một biểu thức trong một dữ liệu-frame

Question

Puzzle cho cognoscenti R: Giả sử chúng ta có một dữ liệu khung:

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

Tôi biết chúng ta có thể làm những việc như

with(df, a) 

để nhận vectơ kết quả.

Nhưng làm cách nào để viết hàm có biểu thức (chẳng hạn như a hoặc a > 3) và thực hiện điều tương tự bên trong. I E. Tôi muốn viết một hàm fn lấy một khung dữ liệu và một biểu thức làm đối số và trả về kết quả của việc đánh giá biểu thức "bên trong" khung dữ liệu dưới dạng môi trường.

Đừng bận tâm rằng điều này nghe có vẻ khó hiểu (tôi chỉ có thể sử dụng with như trên), nhưng đây chỉ là một phiên bản đơn giản của một hàm phức tạp hơn tôi đang viết. Tôi đã thử một số biến thể (sử dụng eval, with, envir, substitute, local, v.v.) nhưng không có biến thể nào hoạt động. Ví dụ nếu tôi xác định fn như vậy:

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

tôi nhận được lỗi này:

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

Rõ ràng tôi đang thiếu một cái gì đó tinh tế về môi trường và đánh giá. Có cách nào để xác định một chức năng như vậy?

Answer 1

Gói mạng thực hiện loại điều này theo một cách khác. Xem, ví dụ: 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

Đó là vì bạn không chuyển biểu thức.

Hãy thử:

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 

Nhìn lướt qua mã nguồn của các chức năng sử dụng này (ví dụ lm) có thể tiết lộ rất nhiều điều thú vị hơn về nó.

Answer 3

? Bên trong cũng có thể được quan tâm.

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

Cách tiếp cận muộn, nhưng cú pháp và phương pháp data.table sẽ xuất hiện như sau. Đây chính xác là cách [.data.table hoạt động với các đối số j, i và by.

Nếu bạn cần nó trong các hình thức fn(x,expr), sau đó bạn có thể sử dụng sau

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 

Tôi nghĩ rằng đó là dễ dàng hơn để sử dụng ở dạng tự nhiên hơn

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

và vân vân. Trong nền, điều này đang sử dụng substitute và eval