R idiom用于矢量查找

Question

这是我写的一个函数:

lookup <- function (keys, values, key, default) {
  found <- which(keys == key)
  if (length(found) == 1) return(values[found])
  if (length(found) == 0) return(default)
  stop("lookup(",keys,",",values,",",key,",",default,"): duplicate keys")
}

它做我需要的就好了:

> lookup(c("a"),c(3),"a",0)
[1] 3
> lookup(c("a"),c(3),"b",0)
[1] 0
> lookup(c("a","a"),c(3),"a",0)
Error in lookup(c("a", "a"), c(3), "a", 0) : lookup(aa,3,a,0): duplicate keys

问题是:它看起来风格正确吗？我错过了什么吗？

(具体来说,我希望这个函数可以写成单个表达式).

我确实理解这涉及完全向量查找,因此效率低下,data.table如果我想要快速,我应该使用它.幸运的是,这种情况下的性能并不重要,因为我的数据非常小.

谢谢!

Answer 1

R通过语言构建了词典 names

dict <- c("Key1"="val1", "Key2"="val2")

dict[["Key1"]]
# [1] "val1"


dict[["Key3"]]
# Error in dict[["Key3"]] : subscript out of bounds

如果您需要一个提供默认值的函数:

getVal <- function(key, dict=defaultDict) { 
  if (! key %in% names(dict) )
    return(defaultValue)

  dict[[key]]
}