r - 生成函数

Question

我对 R 非常陌生，想知道如何创建函数。

如果我有:

> colourName
 [1] "red" "green" "blue" "yellow" "white" "black"

 #Which the following colours equal something like this:
 #red = 1
 #green = 2
 #blue = 3 
 #yellow = 4
 #white = 5
 #black = 6

如何创建一个名为 myColour() 的函数，其中结果以数值向量形式返回？

因此，如果我在下面输入以下内容，我应该得到:

> myColour("yellow")
[1] 4    

请帮忙..

我的代码(但它是错误的!)

colourName<-c("red", "green", "blue", "yellow", "white", "black")
data <- c(1,2,3,4,5,6)

myFunction <- function(colour){
colourName = data
return(colour)
}
myFunction("red")

有没有可能将其创建为函数？

Answer 1

您可以在命名向量中查找您的颜色，您可以像这样高效地执行...

x <- setNames( seq_along( colourName ) , colourName )
#red  green   blue yellow  white  black 
#  1      2      3      4      5      6 

x[ 'red' ]
#red
#  1 

在这里使用函数似乎不是一件好事，但如果你愿意，你可以有一个像这样的简单查找函数，它接受颜色向量和查找值(但你也可以只使用 which!!)...

myFunction <- function( colours , x){
  y <- which( colours %in% x )
  if( length(y) == 0L )
    y <- "Colour not found"
  return( y )
}

myFunction( colourName , "red")
[1] 1

#  Using R's inbuilt colour names
myFunction( colours() , "purple")
[1] 547