如何在R中编写一个接受两个参数的函数

Question

我希望编写一个接受2个参数的函数。

1. 单个正整数“ n”
2. 逻辑指示器“全部”

该函数应执行以下操作：

If TRUE, return the first n prime numbers as a single vector.
ELSE return only the n-th prime number.

输出应为：

genprime(7, all=TRUE)

  

[1] 2 3 5 7 11 13 17

 genprime(7, all=FALSE)

  

[1] 17

Answer 1

让我通过以下示例代码帮助您将结构放置到位：

generateFirstN_primesNumbers <- function(x) {
  #This function returns first x primes numbers
  # DO SOME WORK
  return(finalvector)
}

genprime <- function(n, all = TRUE) {
  allPrimes <-  generateFirstN_primesNumbers(n)
  if (all) {
    allPrimes  
  } else {
    allPrimes[length(allPrimes)]
  }
}

您仍然需要处理generateFirstN_primesNumbers函数，该函数将使用x并返回前X个质数。