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

Question

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

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

我确信这很简单，但我只是不知道该怎么做。

Answer 1

我们可以使用子集

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")