首页 > 解决方案 > R - 创建一个函数 f(x) 作为由 n 索引的和

问题描述

是否可以在 R 中编写一个函数 f(x)

f(x) = a_0 + a_1*sin(x) + ... + a_n*sin(n*x)

对于某些 n 或任何其他 f_i(x) 代替 sin(i*x) 只是在 i 上变化?我尝试了类似的递归

f <- function(x) a_0
for(n in 1:N)
f <- function(x) f(x) + a_n*x^n

它似乎有效,但是当我使用 f(x) 进行一些计算时,R 说嵌套过多。我最终手工编写了 a_0 + a_1*x + ... 等等。有没有一种适当的方法可以在不使用递归的情况下以紧凑的方式完成它?

标签: rfunctionrecursion

解决方案


如果您具有以下值ax

a <- 1:5
x <- 3
a[1] + a[2]*sin(x*1) + a[3]*sin(x*2) + a[4]*sin(x*3) + a[5]*sin(x*4)
# [1] -0.5903971

然后你可以使用

f <- function(x) {
  a[1] + sum( a[-1] * sin((x * seq.int(length(a)-1) )))
}
f(x) 
#[1] -0.5903971

请注意,R 中的数组使用从 1 开始的索引


推荐阅读