r - 用函数评估多项式

Question

我试图评估多项式 P(x)：

但是我不认为我在我的函数中正确输入了多项式本身。功能代码：

directpoly1 <- function(x, coef, seqcoef = seq(coef) - 1) {
  sum(coef*x^seqcoef)
}
directpoly <- function(x, coef) {
  seqcoef <- seq(coef) - 1
  sapply(x, directpoly1, coef, seqcoef)
}

函数使用代码：

directpoly(x=seq(-10,10, length=5000000), rep(c(2,-1),20))

关于如何正确输入的任何想法？

标签： rpolynomials

Answer 1

这是编写函数的两种方法。

1. 带一个sapply循环。使代码更具可读性，并且速度更快。
2. 带有标准for循环。

无需在要使用x值调用的函数之外编写单独的函数，我已将该函数放在主要函数的主体中。
然后用较小的输入向量测试这两个函数。如您所见，for循环功能更快。

directpoly <- function(x, n = 39, coef = NULL){
  f <- function(y) sum(coef * y^seqcoef)
  if(all(is.null(coef))) coef <- rep(c(2, -1), length.out = n + 1)
  seqcoef <- rev(seq_along(coef) - 1)
  sapply(x, f)
}

directpoly2 <- function(x, n = 39, coef = NULL){
  f <- function(y) sum(coef * y^seqcoef)
  if(all(is.null(coef))) coef <- rep(c(2, -1), length.out = n + 1)
  seqcoef <- rev(seq_along(coef) - 1)
  y <- numeric(length(x))
  for(i in seq_along(y))
    y[i] <- f(x[i])
  y
}

library(microbenchmark)
library(ggplot2)

x <- seq(-10, 10, length = 50000)
mb <- microbenchmark(
  Sapply = directpoly(x),
  Forloop = directpoly2(x)
)

autoplot(mb)