r - 如何制作一个在输入相应变量时返回值的函数？

Question

我正在尝试创建一个函数，当我输入“apple”时，它会返回值“1”

data <- data.frame(number = c(1,
                              2,
                              3),
                   fruit = c("apple",
                             "banana",
                             "orange"))

我相信这真的很容易，但我只是不知道该怎么做。

Answer 1

我们可以使用 subset

unlist(subset(data, fruit == "apple", select = 'number'), use.names = FALSE)
#[1] 1

或与 [

data$number[data$fruit == "apple"]

它可以包装成一个函数

f1 <- function(dat, colNm1, colNm2, stringVal) {
         dat[[colNm2]][dat[[colNm1 == stringVal]
   }

f1(data, "fruit", "number", "apple")