Выберите столбец из фрейма данных по заданному имени

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