從給定名稱的數據框中選擇一列

Question

我正在編寫一個函數，在這個函數中，我想做以下事情：

我有一個數據框或列表，我想通過函數中的給定參數選擇特定的列。

示例數據

dat <- cbind.data.frame(A = c(1, 2, 3), B = c(4, 5, 6))

function(dat, name = "B") {
 new_data <- dat[data$name ,]
 return(new_data)
}

我想要得到的基本上是dat$B列。 我的上述功能無法正常工作。 非常感謝你的幫助。

Answer 1

subSET <- function(dat, name) {
 dat[name]
}

subSET2 <- function(dat, name) {
 dat[[name]]
}



subSET(dat, "B") # returns data.frame

  B
1 4
2 5
3 6

subSET2(dat, "B")  # returns vector
[1] 4 5 6